Searched refs:output_to_processing_mapping (Results 1 – 4 of 4) sorted by relevance
80 input_shape.dim_size(shape_spec.output_to_processing_mapping[i])); in EmitDynamicSlice()217 int64 processing_shape_dim = shape_spec.output_to_processing_mapping[i]; in EmitDynamicSlice()329 int64_t input_index = shape_spec.output_to_processing_mapping[i]; in Compile()471 if (shape_spec.output_to_processing_mapping[i] != -1) { in CompileAsDynamicUpdateSlice()472 processing_shape.set_dim(shape_spec.output_to_processing_mapping[i], in CompileAsDynamicUpdateSlice()
389 shape_spec->output_to_processing_mapping.clear(); in ValidateStridedSliceOp()408 shape_spec->output_to_processing_mapping.push_back(gather_index); in ValidateStridedSliceOp()414 shape_spec->output_to_processing_mapping.push_back(-1); in ValidateStridedSliceOp()
40 gtl::InlinedVector<int64_t, 4> output_to_processing_mapping; member
370 processing_dims, shape_spec.output_to_processing_mapping); in Compute()