| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_simplifier.cc | 96 HloScatterInstruction* scatter, in TransformScatterUpdates()
133 auto* scatter = Cast<HloScatterInstruction>(inst); in ExpandInstruction() local
206 if (auto* scatter = DynCast<HloScatterInstruction>(inst)) { in InstructionMatchesPattern() local
|
| D | scatter_expander.cc | 249 HloScatterInstruction* scatter, HloInstruction* induction_var, in ScatterLoopBody()
379 static int64_t ScatterTripCount(const HloScatterInstruction* scatter) { in ScatterTripCount()
414 auto* scatter = Cast<HloScatterInstruction>(inst); in ExpandInstruction() local
488 auto* scatter = DynCast<HloScatterInstruction>(inst); in InstructionMatchesPattern() local
|
| D | hlo_sharding_util.cc | 731 const HloScatterInstruction* scatter) { in ScatterIndexSharding()
774 const HloScatterInstruction* scatter) { in ScatterDataSharding()
816 const HloSharding& index_sharding, const HloScatterInstruction& scatter) { in ScatterEffectiveIndexSharding()
860 const HloScatterInstruction& scatter) { in ScatterEffectiveDataSharding()
1134 const HloSharding& update_sharding, const HloScatterInstruction& scatter) { in ScatterOutputShardingFromUpdate()
1153 const HloScatterInstruction& scatter) { in ScatterUpdateShardingFromOutput()
1173 const HloScatterInstruction& scatter) { in IdentityValueAndHloOpcodeForScatterReduceComputation()
|
| D | dfs_hlo_visitor_with_default.h | 259 Status HandleScatter(HloInstructionPtr scatter) override { in HandleScatter()
|
| D | hlo_cost_analysis.cc | 1078 auto* scatter = Cast<HloScatterInstruction>(hlo); in HandleScatter() local
|
| D | hlo_instructions.h | 1754 HloComputation* scatter() const { in scatter() function
|
| D | hlo_dataflow_analysis_test.cc | 2529 HloInstruction* scatter = computation->root_instruction(); in TEST_F() local
2575 HloInstruction* scatter = computation->root_instruction(); in TEST_F() local
|
| D | hlo_dataflow_analysis.cc | 1773 const HloScatterInstruction* scatter = in GetInPlaceInputOutputPairs() local
|
| D | sharding_propagation.cc | 2182 auto* scatter = Cast<HloScatterInstruction>(instruction); in InferShardingFromOperands() local
|
| D | hlo_verifier.cc | 1662 Status ShapeVerifier::HandleScatter(HloInstruction* scatter) { in HandleScatter()
|
| D | hlo_instruction.cc | 1655 HloComputation* scatter) { in CreateSelectAndScatter()
4773 HloComputation* HloInstruction::scatter() const { in scatter() function in xla::HloInstruction
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 774 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatterPassthroughOperand()
813 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatterIndexPassthroughPartition()
952 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatterTrivialIndexedOperandDimension()
1008 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatter()
1044 auto scatter = Cast<HloScatterInstruction>(hlo); in HandleScatter() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 334 def scatter(self, indices, value, name=None): member in _GraphTensorArray
606 def scatter(self, indices, value, name=None): member in _GraphTensorArrayV2
901 def scatter(self, indices, value, name=None): member in _EagerTensorArray
1255 def scatter(self, indices, value, name=None): member in TensorArray
|
| /external/libdrm/ |
| D | xf86drmHash.c | 85 static unsigned long scatter[256]; in HashHash() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_array_ops_decomposition.cc | 134 } else if (auto scatter = in GetTensorArrayElementShape() local
443 TF::TensorArrayScatterV3Op scatter, in HandleTensorArrayScatterV3Op()
891 } else if (auto scatter = llvm::dyn_cast<TF::TensorArrayScatterV3Op>(&op)) { in DecomposeTensorArrayOps() local
|
| D | tensor_list_ops_decomposition.cc | 778 TF::TensorListScatterIntoExistingListOp scatter, in HandleTensorListScatterIntoExistingListOp()
858 } else if (auto scatter = in DecomposeTensorListOpsInternal() local
|
| D | shape_inference.cc | 319 if (auto scatter = llvm::dyn_cast<TensorListScatterIntoExistingListOp>( in CanInferTensorListElementType() local
|
| /external/mesa3d/src/glx/ |
| D | glxhash.c | 132 static unsigned long scatter[256]; in HashHash() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 405 auto scatter = CreateScalarAddComputation(element_type, ctx->builder()); in Compile() local
|
| /external/llvm/lib/Transforms/Scalar/ |
| D | Scalarizer.cpp | 272 Scatterer Scalarizer::scatter(Instruction *Point, Value *V) { in scatter() function in Scalarizer
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 582 def scatter(operand, scatter_indices, updates, update_computation, function
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Scalar/ |
| D | Scalarizer.cpp | 345 Scatterer ScalarizerVisitor::scatter(Instruction *Point, Value *V) { in scatter() function in ScalarizerVisitor
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 796 auto scatter = Cast<HloScatterInstruction>(instruction); in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 3261 const XlaComputation& scatter) { in SelectAndScatter()
3306 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterInternal()
3336 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
4726 const XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatter()
4737 const XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
|
| /external/rust/crates/ring/src/arithmetic/ |
| D | bigint.rs | 978 fn scatter(table: &mut [Limb], state: &[Limb], i: Window, num_limbs: usize) { in elem_exp_consttime() function
|