| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 225 xla::XlaOp gather; in XlaGatherWithBatchDimsOpImpl() local
253 xla::XlaOp gather; in Compile() local
297 xla::XlaOp gather; in Compile() local
|
| D | variable_ops.cc | 152 xla::XlaOp gather; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_simplifier.cc | 35 auto* gather = DynCast<HloGatherInstruction>(inst); in ExpandInstruction() local
129 auto* gather = DynCast<HloGatherInstruction>(inst); in InstructionMatchesPattern() local
|
| D | gather_expander.cc | 154 const HloInstruction& gather, HloInstruction* induction_var, in GatherLoopBody()
|
| D | hlo_sharding_util.cc | 1012 const HloSharding& output_sharding, const HloInstruction& gather, in GatherParallelDataOperandSharding()
1561 const HloInstruction& gather, const GatherParallelDims& parallel_dim) { in GatherParallelOutputDims()
1581 const HloInstruction& gather, const GatherParallelDims& parallel_dims) { in GatherOutputAlignedOperandParallelDims()
|
| D | dfs_hlo_visitor_with_default.h | 256 Status HandleGather(HloInstructionPtr gather) override { in HandleGather()
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | gather_op_test.py | 319 def gather(x, indices, axis): function
483 def gather(params): function
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 168 const HloGatherInstruction* gather, const Shape& output_shape, in PartitionGatherIndexPassthroughPartition()
271 const HloGatherInstruction* gather, Shape output_shape, in PartitionGatherPassthroughOperand()
375 const HloGatherInstruction* gather, Shape output_shape, in PartitionGatherTrivialIndexedOperandDimension()
486 const HloGatherInstruction* gather, Shape output_shape, in PartitionGatherIndexParallelDimensions()
638 const HloGatherInstruction* gather, PartitionedHlo& operand, in PartitionGather()
686 auto gather = Cast<HloGatherInstruction>(hlo); in HandleGather() local
|
| D | spmd_partitioner_test.cc | 6757 auto gather = op::Gather(op::Parameter(0), op::Subtract(clamp, min)); in TEST_F() local
6788 auto gather = op::Gather(op::Parameter(0), op::Subtract(clamp, min)); in TEST_F() local
9212 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9242 auto gather = AllOf(op::Shape("s32[2,2,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9273 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9303 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9334 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9365 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9396 auto gather = AllOf(op::Shape("s32[4,1,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9430 auto gather = AllOf(op::Shape("s32[4,2,2,2]"), op::Gather(operand, indices)); in TEST_F() local
[all …]
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Gather.cpp | 40 SimpleTensor<T> gather(const SimpleTensor<T> &src, const SimpleTensor<uint32_t> &indices, uint32_t … in gather() function
|
| /external/python/cpython3/Lib/curses/ |
| D | textpad.py | 160 def gather(self): member in Textbox
|
| /external/python/cpython2/Lib/curses/ |
| D | textpad.py | 160 def gather(self): member in Textbox
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | GatherFixture.h | 91 FunctionType gather; in compute_target() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | optimize_mhlo.cc | 55 LogicalResult matchAndRewrite(GatherOp gather, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | test_util.py | 66 def gather(strategy, value): function
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_fn_op_test.py | 261 def gather(x): function
|
| D | ragged_gather_ops.py | 34 def gather(params: ragged_tensor.RaggedOrDense, function
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 296 def gather(self, indices, name=None): member in _GraphTensorArray
568 def gather(self, indices, name=None): member in _GraphTensorArrayV2
865 def gather(self, indices, name=None): member in _EagerTensorArray
1203 def gather(self, indices, name=None): member in TensorArray
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather.cc | 30 namespace gather { namespace
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_array_ops_decomposition.cc | 143 } else if (auto gather = in GetTensorArrayElementShape() local
426 TF::TensorArrayGatherV3Op gather, in HandleTensorArrayGatherV3Op()
889 } else if (auto gather = llvm::dyn_cast<TF::TensorArrayGatherV3Op>(&op)) { in DecomposeTensorArrayOps() local
|
| D | tensor_list_ops_decomposition.cc | 763 TF::TensorListGatherOp gather, in HandleTensorListGatherOp()
854 } else if (auto gather = llvm::dyn_cast<TF::TensorListGatherOp>(&op)) { in DecomposeTensorListOpsInternal() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | value_inference_test.cc | 282 auto gather = Gather(operand1, indices, dim_numbers, {1}); in TEST_F() local
298 auto gather = Gather(data_operand, indices, dim_numbers, {1}); in TEST_F() local
316 auto gather = Gather(operand1, indices, dim_numbers, {1}); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_array_ops.py | 115 def gather(params, function
|
| /external/python/cpython3/Lib/asyncio/ |
| D | tasks.py | 684 def gather(*coros_or_futures, return_exceptions=False): function
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 223 GatherComp gather; in TEST() local
|