| /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 | random_ops.cc | 205 xla::XlaOp gather; in Compile() local
|
| D | variable_ops.cc | 136 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()
459 StatusOr<HloInstruction*> PartitionGather(const HloGatherInstruction* gather, in PartitionGather()
672 auto gather = Cast<HloGatherInstruction>(hlo); in HandleGather() local
|
| D | spmd_partitioner_test.cc | 4540 auto gather = op::Gather(op::Parameter(0), op::Subtract(clamp, min)); in TEST_F() local
4571 auto gather = op::Gather(op::Parameter(0), op::Subtract(clamp, min)); in TEST_F() local
6710 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
6740 auto gather = AllOf(op::Shape("s32[2,2,2,2]"), op::Gather(operand, indices)); in TEST_F() local
6771 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
6801 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
6832 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
6863 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
6894 auto gather = AllOf(op::Shape("s32[4,1,2,2]"), op::Gather(operand, indices)); in TEST_F() local
6924 auto gather = AllOf(op::Shape("s32[2,4,1,2]"), op::Gather(operand, indices)); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | test_util.py | 37 def gather(strategy, value): function
|
| /external/python/cpython2/Lib/curses/ |
| D | textpad.py | 160 def gather(self): member in Textbox
|
| /external/python/cpython3/Lib/curses/ |
| D | textpad.py | 160 def gather(self): member in Textbox
|
| /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/ops/ragged/ |
| D | ragged_gather_ops.py | 36 def gather(params, function
|
| D | ragged_map_fn_op_test.py | 263 def gather(x): function
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamism_inference_test.cc | 285 auto gather = Gather(operand1, indices, dim_numbers, {1}); in TEST_F() local
305 auto gather = Gather(data_operand, indices, dim_numbers, {1}); in TEST_F() local
327 auto gather = Gather(operand1, indices, dim_numbers, {1}); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather.cc | 30 namespace gather { namespace
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | gather.cc | 30 namespace gather { namespace
|
| /external/skqp/src/gpu/ |
| D | GrTextureOpList.cpp | 195 auto gather = [ alloc SkDEBUGCODE(, this) ] (GrSurfaceProxy* p) { in gatherProxyIntervals() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 293 def gather(self, indices, name=None): member in _GraphTensorArray
561 def gather(self, indices, name=None): member in _GraphTensorArrayV2
856 def gather(self, indices, name=None): member in _EagerTensorArray
1172 def gather(self, indices, name=None): member in TensorArray
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 151 const HloInstruction& gather, HloInstruction* induction_var, in GatherLoopBody()
|
| D | hlo_sharding_util.cc | 895 const HloSharding& output_sharding, const HloInstruction& gather, in GatherParallelDataOperandSharding()
1338 const HloInstruction& gather, const GatherParallelDims& parallel_dim) { in GatherParallelOutputDims()
1358 const HloInstruction& gather, const GatherParallelDims& parallel_dims) { in GatherOutputAlignedOperandParallelDims()
|
| /external/skqp/src/shaders/ |
| D | SkImageShader.cpp | 314 auto gather = alloc->make<SkRasterPipeline_GatherCtx>(); in onAppendStages() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_array_ops_decomposition.cc | 144 } else if (auto gather = in GetTensorArrayElementShape() local
427 TF::TensorArrayGatherV3Op gather, in HandleTensorArrayGatherV3Op()
886 } else if (auto gather = llvm::dyn_cast<TF::TensorArrayGatherV3Op>(&op)) { in DecomposeTensorArrayOps() local
|
| D | tensor_list_ops_decomposition.cc | 773 TF::TensorListGatherOp gather, in HandleTensorListGatherOp()
864 } else if (auto gather = llvm::dyn_cast<TF::TensorListGatherOp>(&op)) { in DecomposeTensorListOpsInternal() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 91 def gather(operand, start_indices): function
|
| /external/swiftshader/src/Reactor/ |
| D | EmulatedIntrinsics.cpp | 60 void gather(T &out, RValue<Pointer<EL>> base, RValue<Int4> offsets, RValue<Int4> mask, unsigned int… in gather() function
|
| /external/python/cpython3/Lib/asyncio/ |
| D | tasks.py | 720 def gather(*coros_or_futures, loop=None, return_exceptions=False): function
|
| /external/swiftshader/src/Renderer/ |
| D | Sampler.hpp | 229 bool gather; member in sw::Sampler
|