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.cc942 int64_t input_dim_start = -1; in HandleReshape() local
953 input_dim_start = start.first; in HandleReshape()
959 if ((input_dim_end - input_dim_start) > 1 && in HandleReshape()
1022 int64_t input_dim_start = -1; in HandleReshape() local
1033 input_dim_start = start.first; in HandleReshape()
1040 VLOG(2) << "Input dim start: " << input_dim_start in HandleReshape()
1045 if ((input_dim_end - input_dim_start) > 1 && in HandleReshape()
1055 if (common_factor.first == input_dim_start) { in HandleReshape()
Dshape_inference.cc3126 int64_t input_dim_start = -1; in InferReshapeShape() local
3135 input_dim_start = start.first; in InferReshapeShape()
3142 if ((input_dim_end - input_dim_start) > 1 && in InferReshapeShape()
3175 if (common_factor.first == input_dim_start) { in InferReshapeShape()