Searched refs:input_num_elements (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_op.h | 71 int64 input_num_elements = 1; in Compute() local
78 input_num_elements *= input.dim_size(dim); in Compute()
84 const int64 missing = input_num_elements / product; in Compute()
87 context, product * missing == input_num_elements, in Compute()
89 "Input to reshape is a tensor with ", input_num_elements, in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reshape_op.cc | 81 int64 input_num_elements = 1; in Compile() local
88 input_num_elements *= input_shape.dim_size(dim); in Compile()
94 const int64 missing = input_num_elements / product; in Compile()
97 ctx, product * missing == input_num_elements, in Compile()
99 "Input to reshape is a tensor with ", input_num_elements, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_reshape_op.cc | 102 int64 input_num_elements = 1; in Compute() local
109 input_num_elements *= input_shape.dim_size(dim); in Compute()
115 const int64 missing = input_num_elements / product; in Compute()
118 context, product * missing == input_num_elements, in Compute()
120 "Input to reshape is a tensor with ", input_num_elements, in Compute()
|
| /external/XNNPACK/test/ |
| D | resize-bilinear-operator-tester.h | 323 const int64_t input_num_elements = input_num_pixels * input_pixel_stride(); in TestNCHWxF32() local
339 …input[batch_index * input_num_elements + c * input_num_pixels + input_y_top * input_width() + inpu… in TestNCHWxF32()
340 …input[batch_index * input_num_elements + c * input_num_pixels + input_y_top * input_width() + inpu… in TestNCHWxF32()
341 …input[batch_index * input_num_elements + c * input_num_pixels + input_y_bottom * input_width() + i… in TestNCHWxF32()
342 …input[batch_index * input_num_elements + c * input_num_pixels + input_y_bottom * input_width() + i… in TestNCHWxF32()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2771 const int input_num_elements = in IsReductionCandidateForSimplification() local
2775 *is_single_element_op = input_num_elements == 1 && output_num_elements == 1; in IsReductionCandidateForSimplification()
|