Home
last modified time | relevance | path

Searched refs:input_dim_end (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc833 int64 input_dim_end = -1; in HandleReshape() local
844 input_dim_end = end.first; in HandleReshape()
851 << " Input dim end: " << input_dim_end in HandleReshape()
855 if ((input_dim_end - input_dim_start) > 1 && in HandleReshape()
884 if (common_factor.first == input_dim_end) { in HandleReshape()
Dshape_inference.cc2987 int64 input_dim_end = -1; in InferReshapeShape() local
2996 input_dim_end = end.first; in InferReshapeShape()
3002 if ((input_dim_end - input_dim_start) > 1 && in InferReshapeShape()
3038 if (common_factor.first == input_dim_end) { in InferReshapeShape()