Searched defs:new_indices (Results 1 – 8 of 8) sorted by relevance
| /external/ComputeLibrary/src/core/CPP/kernels/ |
| D | CPPBoxWithNonMaximaSuppressionLimitKernel.cpp | 148 std::vector<int> new_indices; in NonMaximaSuppression() local
|
| /external/ComputeLibrary/src/runtime/CPP/functions/ |
| D | CPPDetectionOutputLayer.cpp | 543 std::map<int, std::vector<int>> new_indices; in run() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2999 std::vector<HloInstruction*> new_indices(hlo->shape().rank()); in HandleDynamicSlice() local
3063 std::vector<HloInstruction*> new_indices(hlo->shape().rank()); in HandleDynamicUpdateSlice() local
3089 std::vector<HloInstruction*> new_indices(hlo->shape().rank()); in HandleDynamicUpdateSlice() local
3206 std::vector<HloInstruction*> new_indices(hlo->shape().rank()); in HandleDynamicUpdateSlice() local
|
| D | gather_scatter_handler.cc | 420 PartitionedHlo new_indices(adjusted_indices, adjusted_indices->shape(), in PartitionGatherTrivialIndexedOperandDimension() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | meta_spmd_expander.cc | 1071 mlir::Value new_indices = builder.create<mlir::TF::SubOp>( in ExpandOp() local
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 613 def _process_body(self, pfor_input, inputs_stacked, new_indices, cond_stacked, argument
4845 def _process_body(self, inputs_stacked, new_indices, cond_stacked, argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 1032 DimensionVector new_indices(result_shape.rank()); in SliceInternal() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 4819 std::vector<HloInstruction*> new_indices; in HandleDynamicSlice() local
|