Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc832 int64 input_dim_start = -1; in HandleReshape() local
843 input_dim_start = start.first; in HandleReshape()
850 VLOG(2) << "Input dim start: " << input_dim_start in HandleReshape()
855 if ((input_dim_end - input_dim_start) > 1 && in HandleReshape()
881 if (common_factor.first == input_dim_start) { in HandleReshape()
Dshape_inference.cc2986 int64 input_dim_start = -1; in InferReshapeShape() local
2995 input_dim_start = start.first; in InferReshapeShape()
3002 if ((input_dim_end - input_dim_start) > 1 && in InferReshapeShape()
3035 if (common_factor.first == input_dim_start) { in InferReshapeShape()