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.cc2322 const Shape& first_index_shape = start_index_shapes[0]; in InferDynamicSliceShape() local
2323 if (!ShapeUtil::IsScalar(first_index_shape)) { in InferDynamicSliceShape()
2325 ShapeUtil::HumanString(first_index_shape)); in InferDynamicSliceShape()
2327 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicSliceShape()
2332 if (!ShapeUtil::Compatible(first_index_shape, index_shape)) { in InferDynamicSliceShape()
2336 ShapeUtil::HumanString(first_index_shape), in InferDynamicSliceShape()
2428 const Shape& first_index_shape = start_index_shapes[0]; in InferDynamicUpdateSliceShape() local
2429 if (!ShapeUtil::IsScalar(first_index_shape)) { in InferDynamicUpdateSliceShape()
2432 ShapeUtil::HumanString(first_index_shape)); in InferDynamicUpdateSliceShape()
2434 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicUpdateSliceShape()
[all …]