| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 221 xla::XlaOp gather; in XlaGatherWithBatchDimsOpImpl() local
249 xla::XlaOp gather; in Compile() local
293 xla::XlaOp gather; in Compile() local
|
| D | variable_ops.cc | 137 xla::XlaOp gather; in Compile() local
|
| D | tensor_array_ops.cc | 339 xla::XlaOp gather; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 137 const HloGatherInstruction* gather, absl::Span<const int64> batch_dims, in PartitionIndexOnlyPartition()
180 const HloGatherInstruction* gather, Shape output_shape, in ParititonPassthroughOperand()
215 const HloGatherInstruction* gather, Shape output_shape, in ParititonTrivialIndexedOperandDimension()
315 const HloGatherInstruction* gather, Shape output_shape, in PartitionIndexParallelDimensions()
475 StatusOr<HloInstruction*> PartitionGather(const HloGatherInstruction* gather, in PartitionGather()
694 auto gather = Cast<HloGatherInstruction>(hlo); in HandleGather() local
|
| D | spmd_partitioner_test.cc | 4917 auto gather = op::Gather(op::Parameter(0), op::Subtract(clamp, min)); in TEST_F() local
4948 auto gather = op::Gather(op::Parameter(0), op::Subtract(clamp, min)); in TEST_F() local
7200 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
7230 auto gather = AllOf(op::Shape("s32[2,2,2,2]"), op::Gather(operand, indices)); in TEST_F() local
7261 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
7291 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
7322 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
7353 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
7384 auto gather = AllOf(op::Shape("s32[4,1,2,2]"), op::Gather(operand, indices)); in TEST_F() local
7415 auto gather = AllOf(op::Shape("s32[2,4,1,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/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | optimize_mhlo.cc | 57 LogicalResult matchAndRewrite(GatherOp gather, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | test_util.py | 45 def gather(strategy, value): function
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_ops.py | 36 def gather(params, function
|
| D | ragged_map_fn_op_test.py | 265 def gather(x): function
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 300 def gather(self, indices, name=None): member in _GraphTensorArray
572 def gather(self, indices, name=None): member in _GraphTensorArrayV2
869 def gather(self, indices, name=None): member in _EagerTensorArray
1200 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/xla/service/ |
| D | gather_expander.cc | 152 const HloInstruction& gather, HloInstruction* induction_var, in GatherLoopBody()
|
| D | hlo_sharding_util.cc | 928 const HloSharding& output_sharding, const HloInstruction& gather, in GatherParallelDataOperandSharding()
1438 const HloInstruction& gather, const GatherParallelDims& parallel_dim) { in GatherParallelOutputDims()
1458 const HloInstruction& gather, const GatherParallelDims& parallel_dims) { in GatherOutputAlignedOperandParallelDims()
|
| D | dfs_hlo_visitor_with_default.h | 249 Status HandleGather(HloInstructionPtr gather) override { in HandleGather()
|
| /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()
885 } else if (auto gather = llvm::dyn_cast<TF::TensorArrayGatherV3Op>(&op)) { in DecomposeTensorArrayOps() local
|
| D | tensor_list_ops_decomposition.cc | 757 TF::TensorListGatherOp gather, in HandleTensorListGatherOp()
848 } else if (auto gather = llvm::dyn_cast<TF::TensorListGatherOp>(&op)) { in DecomposeTensorListOpsInternal() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | value_inference_test.cc | 256 auto gather = Gather(operand1, indices, dim_numbers, {1}); in TEST_F() local
272 auto gather = Gather(data_operand, indices, dim_numbers, {1}); in TEST_F() local
290 auto gather = Gather(operand1, indices, dim_numbers, {1}); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_array_ops.py | 103 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 | 206 GatherComp gather; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 92 def gather(operand, start_indices): function
|