/external/tensorflow/tensorflow/core/kernels/ |
shape_ops.h | 133 const int64 size = shape.num_elements();
|
transpose_op.cc | 200 if (shape.num_elements() > 0) {
|
fused_batch_norm_op.cc | 244 if (x.shape().num_elements() == 0) { 649 if (x.shape().num_elements() == 0) { [all...] |
mkl_conv_ops.cc | 189 if (out_shape.num_elements() == 0) { 549 if (output_tf_shape.num_elements() == 0 || output_dims_tf_order[0] == 0) { [all...] |
pooling_ops_common.h | 331 ksize.resize(tensor_ksize.shape().num_elements()); 336 stride.resize(tensor_stride.shape().num_elements());
|
slice_op.cc | 175 if (output_shape.num_elements() > 0) { 250 if (output_shape.num_elements() > 0) {
|
conv_grad_input_ops.cc | 245 if (input_shape.num_elements() == 0) { 360 if (input_shape.num_elements() == 0) { 671 if (input_shape.num_elements() == 0) { [all...] |
/external/tensorflow/tensorflow/tools/graph_transforms/ |
quantize_nodes_test.cc | 249 const size_t x_num_elements = TensorShape(x_shape).num_elements(); 256 const size_t y_num_elements = TensorShape(y_shape).num_elements(); 284 const size_t x_num_elements = TensorShape(x_shape).num_elements(); 291 const size_t y_num_elements = TensorShape(y_shape).num_elements(); [all...] |
/hardware/intel/common/libmix/videodecoder/ |
VideoDecoderMPEG2.cpp | 282 1, //num_elements
|
/hardware/intel/img/psb_video/src/ |
pnw_MPEG2.c | 716 ASSERT(obj_buffer->num_elements == 1); 719 if ((obj_buffer->num_elements != 1) || 868 ASSERT(obj_buffer->num_elements == 1); 871 if ((obj_buffer->num_elements != 1) || [all...] |
/external/mesa3d/src/gallium/drivers/etnaviv/ |
etnaviv_emit.c | 253 size += ctx->vertex_elements->num_elements + 1; 333 ctx->vertex_elements->num_elements, [all...] |
/external/mesa3d/src/gallium/drivers/nouveau/nvc0/ |
nvc0_vbo_translate.c | 51 nvc0->vertprog->vp.need_vertex_id && (nvc0->vertex->num_elements < 32); 631 unsigned a = nvc0->vertex->num_elements;
|
/external/mesa3d/src/gallium/drivers/virgl/ |
virgl_context.c | 340 unsigned num_elements, 346 num_elements, elements); [all...] |
/external/mesa3d/src/gallium/tools/trace/ |
dump_state.py | 438 def create_vertex_elements_state(self, num_elements, elements): 439 return elements[0:num_elements]
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
pooling_ops.cc | 90 OP_REQUIRES(ctx, ksize_shape.num_elements() == num_dims(), 102 OP_REQUIRES(ctx, stride_shape.num_elements() == num_dims(),
|
tensor_array_ops.cc | 526 OP_REQUIRES(ctx, value_shape.num_elements() == ta_shape.num_elements(),
|
/external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
mixture.py | 339 samples_size = static_samples_shape.num_elements() 346 batch_size = static_batch_shape.num_elements()
|
vector_diffeomixture.py | 552 batch_size = self.batch_shape.num_elements() 555 mix_batch_size = self.mixture_distribution.batch_shape.num_elements() 579 2)[-2:].num_elements() [all...] |
/external/tensorflow/tensorflow/python/debug/cli/ |
tensor_format.py | 271 num_elements = line_content.count(",") 272 curr_indices[curr_dim - 1] += num_elements
|
/external/tensorflow/tensorflow/python/ops/ |
data_flow_ops.py | [all...] |
gradients_impl.py | 91 num_elements = np.prod(dense_shape_value) 92 if num_elements >= _LARGE_SPARSE_NUM_ELEMENTS: 95 "This may consume a large amount of memory." % num_elements) [all...] |
/hardware/intel/img/psb_video/src/mrst/ |
psb_MPEG4.c | [all...] |
/external/mesa3d/src/compiler/spirv/ |
spirv_to_nir.c | 919 c->num_elements = glsl_get_length(type); 920 c->elements = ralloc_array(b, nir_constant *, c->num_elements); 923 for (unsigned i = 1; i < c->num_elements; i++) 928 c->num_elements = glsl_get_length(type); 929 c->elements = ralloc_array(b, nir_constant *, c->num_elements); 931 for (unsigned i = 0; i < c->num_elements; i++) { [all...] |
/external/tensorflow/tensorflow/python/feature_column/ |
feature_column.py | 200 num_elements = column._variable_shape.num_elements() # pylint: disable=protected-access 203 array_ops.reshape(tensor, shape=(batch_size, num_elements))) [all...] |
/external/libxaac/test/ |
ixheaacd_main.c | 1396 WORD32 num_elements; local [all...] |