Home
last modified time | relevance | path

Searched defs:scatter (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_select_and_scatter_op.cc94 XlaCompiler::CompilationResult scatter; in Compile() local
Dpooling_ops.cc336 auto scatter = CreateScalarAddComputation(element_type, ctx->builder()); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dscatter_expander.cc220 HloInstruction* scatter, HloInstruction* induction_var, in ScatterLoopBody()
346 HloInstruction* scatter) { in ExpandScatter()
Ddfs_hlo_visitor_with_default.h209 Status HandleScatter(HloInstructionPtr scatter) override { in HandleScatter()
Dhlo_instructions.h1133 HloComputation* scatter() const { in scatter() function
Dhlo_cost_analysis.cc720 Status HloCostAnalysis::HandleScatter(const HloInstruction* scatter) { in HandleScatter()
Dhlo_evaluator_typed_visitor.h1880 HloComputation* scatter = select_and_scatter->scatter(); in HandleSelectAndScatter() local
2307 Status HandleScatter(HloInstruction* scatter) override { in HandleScatter()
Dhlo_verifier.cc818 Status ShapeVerifier::HandleScatter(HloInstruction* scatter) { in HandleScatter()
Dtuple_points_to_analysis_test.cc1057 HloInstruction* scatter = computation_->root_instruction(); in TEST_F() local
Dhlo_instruction.cc1095 HloComputation* scatter) { in CreateSelectAndScatter()
3488 HloComputation* HloInstruction::scatter() const { in scatter() function in xla::HloInstruction
Dhlo_dataflow_analysis_test.cc2436 HloInstruction* scatter = computation_->root_instruction(); in TEST_F() local
Dhlo_parser.cc1234 optional<HloComputation*> scatter; in ParseInstructionRhs() local
Dhlo_instructions.cc1998 HloComputation* scatter) in HloSelectAndScatterInstruction()
/external/tensorflow/tensorflow/python/ops/
Dtensor_array_ops.py342 def scatter(self, indices, value, name=None): member in _GraphTensorArray
632 def scatter(self, indices, value, name=None): member in _GraphTensorArrayV2
906 def scatter(self, indices, value, name=None): member in _EagerTensorArray
1198 def scatter(self, indices, value, name=None): member in TensorArray
/external/libdrm/
Dxf86drmHash.c84 static unsigned long scatter[256]; in HashHash() local
/external/mesa3d/src/glx/
Dglxhash.c132 static unsigned long scatter[256]; in HashHash() local
/external/llvm/lib/Transforms/Scalar/
DScalarizer.cpp272 Scatterer Scalarizer::scatter(Instruction *Point, Value *V) { in scatter() function in Scalarizer
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Transforms/Scalar/
DScalarizer.cpp306 Scatterer Scalarizer::scatter(Instruction *Point, Value *V) { in scatter() function in Scalarizer
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2250 const XlaComputation& scatter) { in SelectAndScatter()
2266 const XlaOp& init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
3263 const XlaComputation& scatter) { in SelectAndScatter()
3274 const XlaOp& init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc555 const LocalOp& init_value, const Computation& scatter) { in SelectAndScatterWithGeneralPadding()
Dxla_client.py1283 padding, source, init_value, scatter): argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1029 Status IrEmitterUnnested::HandleScatter(HloInstruction* scatter) { in HandleScatter()
1075 Thunk* thunk, HloInstruction* scatter, in EmitScatter()