Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc2530 const Shape& first_index_shape = start_index_shapes[0]; in InferDynamicSliceShape() local
2531 if (!ShapeUtil::IsScalar(first_index_shape)) { in InferDynamicSliceShape()
2533 ShapeUtil::HumanString(first_index_shape)); in InferDynamicSliceShape()
2535 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicSliceShape()
2540 if (!ShapeUtil::Compatible(first_index_shape, index_shape)) { in InferDynamicSliceShape()
2544 ShapeUtil::HumanString(first_index_shape), in InferDynamicSliceShape()
2636 const Shape& first_index_shape = start_index_shapes[0]; in InferDynamicUpdateSliceShape() local
2637 if (!ShapeUtil::IsScalar(first_index_shape)) { in InferDynamicUpdateSliceShape()
2640 ShapeUtil::HumanString(first_index_shape)); in InferDynamicUpdateSliceShape()
2642 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicUpdateSliceShape()
[all …]