Searched refs:wrapped_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 166 auto wrapped_index = xla::Add(dim_size, begin_index); in EmitDynamicSlice() local
168 begin_index = xla::Select(index_negative, wrapped_index, begin_index); in EmitDynamicSlice()
178 auto wrapped_index = xla::Add(dim_size, end_index); in EmitDynamicSlice() local
179 end_index = xla::Select(index_negative, wrapped_index, end_index); in EmitDynamicSlice()
458 auto wrapped_index = xla::Add(dim_size, begin_index); in CompileAsDynamicUpdateSlice() local
460 begin_index = xla::Select(index_negative, wrapped_index, begin_index); in CompileAsDynamicUpdateSlice()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 583 int32 wrapped_index = reduction_index; in ProcessTensorFlowReductionOperator() local
584 if (wrapped_index < 0) { in ProcessTensorFlowReductionOperator()
585 wrapped_index += input_rank; in ProcessTensorFlowReductionOperator()
587 true_indices.insert(wrapped_index); in ProcessTensorFlowReductionOperator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1818 auto wrapped_index = reduction_index; in ReductionShapeHelper() local
1819 if (wrapped_index < 0) { in ReductionShapeHelper()
1820 wrapped_index += input_rank; in ReductionShapeHelper()
1823 true_indices->insert(wrapped_index); in ReductionShapeHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3270 auto wrapped_index = in rewriteWithUnknownBegin() local
3273 loc, type, index_negative, wrapped_index, reshaped_index); in rewriteWithUnknownBegin()
|