Home
last modified time | relevance | path

Searched defs:start_indices_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc31 const Shape& start_indices_shape = start_indices->shape(); in TransposeIndexVectorDimToLast() local
87 const Shape& start_indices_shape, HloInstruction* accumulator, in AdjustBatchDimsInAccumulator()
314 const Shape& start_indices_shape = start_indices->shape(); in ExpandInstruction() local
Dshape_inference.cc2279 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicSliceShape() local
2384 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicUpdateSliceShape() local
2867 const Shape& input_shape, absl::Span<const int64> start_indices_shape, in ValidateGatherDimensionNumbers()
2958 const Shape& input_shape, const Shape& start_indices_shape, in InferGatherShape()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddynamic_slice_ops.cc76 const TensorShape start_indices_shape = ctx->InputShape("start_indices"); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc758 auto start_indices_shape = ShapeUtil::MakeShape(S32, {}); in BM_DynamicSlice() local