Searched refs:input_num_elements (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reshape_op.cc | 84 int64_t input_num_elements = 1; in Compile() local
91 input_num_elements *= input_shape.dim_size(dim); in Compile()
97 int64_t missing = input_num_elements / product; in Compile()
101 ctx, product * missing == input_num_elements, in Compile()
103 "Input to reshape is a tensor with ", input_num_elements, in Compile()
112 xla::CeilOfRatio(input_num_elements, product) * product; in Compile()
116 0, 0, padded_input_num - input_num_elements); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_op.h | 73 int64_t input_num_elements = 1; in Compute() local
80 input_num_elements *= input.dim_size(dim); in Compute()
86 const int64_t missing = input_num_elements / product; in Compute()
89 context, product * missing == input_num_elements, in Compute()
91 "Input to reshape is a tensor with ", input_num_elements, in Compute()
|
| /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 | 489 const int64_t input_num_elements = input_num_pixels * input_pixel_stride(); in TestNCHWxF32() local
505 …input[batch_index * input_num_elements + c * input_num_pixels + input_y_top * input_width() + inpu… in TestNCHWxF32()
506 …input[batch_index * input_num_elements + c * input_num_pixels + input_y_top * input_width() + inpu… in TestNCHWxF32()
507 …input[batch_index * input_num_elements + c * input_num_pixels + input_y_bottom * input_width() + i… in TestNCHWxF32()
508 …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 | 2787 const int input_num_elements = in IsReductionCandidateForSimplification() local
2791 *is_single_element_op = input_num_elements == 1 && output_num_elements == 1; in IsReductionCandidateForSimplification()
|