Searched refs:num_elements (Results 151 - 175 of 335) sorted by relevance

1234567891011>>

/external/tensorflow/tensorflow/python/training/
H A Dinput_test.py759 self.assertIs(None, sparse.dense_shape.get_shape().num_elements())
761 self.assertIs(None, batched.dense_shape.get_shape().num_elements())
768 self.assertIs(None, sparse.dense_shape.get_shape().num_elements())
770 self.assertIs(None, batched.dense_shape.get_shape().num_elements())
882 self.assertIs(None, sparse.dense_shape.get_shape().num_elements())
884 self.assertIs(None, batched.dense_shape.get_shape().num_elements())
891 self.assertIs(None, sparse.dense_shape.get_shape().num_elements())
894 self.assertIs(None, batched.dense_shape.get_shape().num_elements())
901 self.assertIs(None, sparse.dense_shape.get_shape().num_elements())
904 self.assertIs(None, batched.dense_shape.get_shape().num_elements())
[all...]
/external/mesa3d/src/gallium/state_trackers/va/
H A Dpicture_hevc.c35 assert(buf->size >= sizeof(VAPictureParameterBufferHEVC) && buf->num_elements == 1);
183 assert(buf->size >= sizeof(VAIQMatrixBufferH264) && buf->num_elements == 1);
196 assert(buf->size >= sizeof(VASliceParameterBufferHEVC) && buf->num_elements == 1);
/external/tensorflow/tensorflow/core/framework/
H A Dtensor_shape.h69 int64 num_elements() const { return num_elements_; } function in class:tensorflow::TensorShapeRep
248 bool IsFullyDefined() const { return !kIsPartial || num_elements() != -1; }
387 static Status NumElements(gtl::ArraySlice<int64> shape, int64* num_elements);
/external/tensorflow/tensorflow/core/kernels/
H A Dmaxpooling_op.cc261 ksize.resize(tensor_ksize.shape().num_elements());
266 stride.resize(tensor_stride.shape().num_elements());
385 ksize.resize(tensor_ksize.shape().num_elements());
390 stride.resize(tensor_stride.shape().num_elements());
490 ksize.resize(tensor_ksize.shape().num_elements());
495 stride.resize(tensor_stride.shape().num_elements());
691 ksize.resize(tensor_ksize.shape().num_elements());
696 stride.resize(tensor_stride.shape().num_elements());
830 ksize.resize(tensor_ksize.shape().num_elements());
835 stride.resize(tensor_stride.shape().num_elements());
[all...]
H A Dpadding_fifo_queue.cc69 void PaddingFIFOQueue::TryDequeueMany(int num_elements, OpKernelContext* ctx, argument
72 if (num_elements == 0) {
99 num_elements, [callback]() { callback(Tuple()); }, ctx, cm, token,
H A Dpriority_queue.cc238 void PriorityQueue::TryDequeueMany(int num_elements, OpKernelContext* ctx, argument
248 if (num_elements == 0) {
300 num_elements, [callback]() { callback(Tuple()); }, ctx, cm, token,
H A Dextract_image_patches_op.cc103 if (out_shape.num_elements() == 0) {
H A Dhistogram_op.cc86 OP_REQUIRES(ctx, (value_range_tensor.shape().num_elements() == 2),
H A Dlist_kernels.cc268 c, c->input(0).shape().num_elements() == 1,
358 int32 num_elements = c->input(1).scalar<int32>()(); variable
362 output.tensors.resize(num_elements, Tensor(DT_INVALID));
382 .HostMemory("num_elements"),
395 c, c->input(0).shape().num_elements() == 1,
H A Dmkl_conv_ops.h386 if (input_tf_shape.num_elements() == 0 ||
387 filter_tf_shape.num_elements() == 0 ||
388 outbprop_tf_shape.num_elements() == 0) {
399 for (size_t i = 0; i < output_tf_shape.num_elements(); ++i) {
H A Drandom_shuffle_queue_op.cc57 void TryDequeueMany(int num_elements, OpKernelContext* ctx,
274 void RandomShuffleQueue::TryDequeueMany(int num_elements, OpKernelContext* ctx, argument
284 if (num_elements == 0) {
336 num_elements, [callback]() { callback(Tuple()); }, ctx, cm, token,
H A Dstateless_random_ops.cc50 if (shape.num_elements() == 0) return;
H A Ddebug_ops.h236 // Use DT_INT64/int64 to be consistent with TensorShape::num_elements().
245 for (int64 i = 0; i < input_shape.num_elements(); ++i) {
301 element_count = input_shape.num_elements();
H A Dtensor_array_ops.cc646 if (output_shape.num_elements() == 0) {
653 output_tensor->shaped<T, 2>({1, output_shape.num_elements()});
899 if (output_shape.num_elements() > 0) {
901 output_tensor->shaped<T, 2>({1, output_shape.num_elements()});
1087 {1, num_values, element_shape.num_elements()});
1091 element_shape.num_elements()};
1103 tensor_value_i->shaped<T, 3>({1, 1, element_shape.num_elements()});
1106 if (element_shape.num_elements() > 0) {
/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsummarize_graph_main.cc218 const size_t num_elements = tensor.NumElements(); local
220 const_parameter_count += num_elements;
222 variable_parameter_count += num_elements;
/external/mesa3d/src/gallium/drivers/etnaviv/
H A Detnaviv_internal.h199 unsigned num_elements; member in struct:compiled_vertex_elements_state
/external/mesa3d/src/gallium/drivers/radeonsi/
H A Dsi_debug.c368 unsigned num_elements,
381 for (i = 0; i < num_elements; i++) {
454 unsigned num_elements[] = { local
468 num_elements[i], f);
365 si_dump_descriptor_list(struct si_descriptors *desc, const char *shader_name, const char *elem_name, unsigned num_elements, FILE *f) argument
/external/mesa3d/src/gallium/drivers/vc4/
H A Dvc4_draw.c151 uint32_t num_elements_emit = MAX2(vtx->num_elements, 1);
185 for (int i = 0; i < vtx->num_elements; i++) {
211 if (vtx->num_elements == 0) {
226 assert(vtx->num_elements <= 8);
H A Dvc4_state.c354 vc4_vertex_state_create(struct pipe_context *pctx, unsigned num_elements, argument
362 memcpy(so->pipe, elements, sizeof(*elements) * num_elements);
363 so->num_elements = num_elements;
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbcast_ops.cc100 for (int64 i = 0; i < in_shape.num_elements(); ++i) {
H A Ddiag_op.cc108 int64 size = input_shape.num_elements();
174 int64 size = input_shape.num_elements();
/external/tensorflow/tensorflow/tools/benchmark/
H A Dbenchmark_model.cc203 int64 output_count = output_shape.num_elements();
216 int64 output_count = output_shape.num_elements();
223 int64 output_count = output_shape.num_elements();
605 int64 total_size = inputs[0].shape.num_elements();
/external/webp/include/webp/
H A Dmux.h369 // num_elements - (out) number of chunks with the given chunk id
371 // WEBP_MUX_INVALID_ARGUMENT - if mux, or num_elements is NULL.
374 WebPChunkId id, int* num_elements);
/external/webp/src/mux/
H A Dmuxread.c523 WebPChunkId id, int* num_elements) {
524 if (mux == NULL || num_elements == NULL) {
529 *num_elements = MuxImageCount(mux->images_, id);
533 *num_elements = CountChunks(*chunk_list, kChunks[idx].tag);
522 WebPMuxNumChunks(const WebPMux* mux, WebPChunkId id, int* num_elements) argument
/external/webp/src/webp/
H A Dmux.h369 // num_elements - (out) number of chunks with the given chunk id
371 // WEBP_MUX_INVALID_ARGUMENT - if mux, or num_elements is NULL.
374 WebPChunkId id, int* num_elements);

Completed in 712 milliseconds

1234567891011>>