Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc384 shape_spec->output_to_processing_mapping.clear(); in ValidateStridedSliceOp()
403 shape_spec->output_to_processing_mapping.push_back(gather_index); in ValidateStridedSliceOp()
409 shape_spec->output_to_processing_mapping.push_back(-1); in ValidateStridedSliceOp()
Dstrided_slice_op.h40 gtl::InlinedVector<int64, 4> output_to_processing_mapping; member
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc79 input_shape.dim_size(shape_spec.output_to_processing_mapping[i])); in EmitDynamicSlice()
292 int64 input_index = shape_spec.output_to_processing_mapping[i]; in Compile()
434 if (shape_spec.output_to_processing_mapping[i] != -1) { in CompileAsDynamicUpdateSlice()
435 processing_shape.set_dim(shape_spec.output_to_processing_mapping[i], in CompileAsDynamicUpdateSlice()