Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_select_and_scatter_op.cc93 XlaCompiler::CompilationResult scatter; in Compile() local
Dpooling_ops.cc349 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.h222 Status HandleScatter(HloInstructionPtr scatter) override { in HandleScatter()
Dhlo_instructions.h1213 HloComputation* scatter() const { in scatter() function
Dhlo_cost_analysis.cc937 Status HloCostAnalysis::HandleScatter(const HloInstruction* scatter) { in HandleScatter()
Dhlo_verifier.cc1001 Status ShapeVerifier::HandleScatter(HloInstruction* scatter) { in HandleScatter()
Dhlo_instruction.cc1190 HloComputation* scatter) { in CreateSelectAndScatter()
3830 HloComputation* HloInstruction::scatter() const { in scatter() function in xla::HloInstruction
Dhlo_parser.cc1296 optional<HloComputation*> scatter; in ParseInstructionRhs() local
Dhlo_dataflow_analysis_test.cc2505 HloInstruction* scatter = computation_->root_instruction(); in TEST_F() local
Dalgebraic_simplifier.cc3969 Status AlgebraicSimplifierVisitor::HandleScatter(HloInstruction* scatter) { in HandleScatter()
Dhlo_instructions.cc2121 HloComputation* scatter) in HloSelectAndScatterInstruction()
/external/libdrm/
Dxf86drmHash.c85 static unsigned long scatter[256]; in HashHash() local
/external/tensorflow/tensorflow/python/ops/
Dtensor_array_ops.py330 def scatter(self, indices, value, name=None): member in _GraphTensorArray
598 def scatter(self, indices, value, name=None): member in _GraphTensorArrayV2
890 def scatter(self, indices, value, name=None): member in _EagerTensorArray
1224 def scatter(self, indices, value, name=None): member in TensorArray
/external/mesa3d/src/glx/
Dglxhash.c132 static unsigned long scatter[256]; in HashHash() local
/external/swiftshader/src/Reactor/
DEmulatedReactor.cpp84 void scatter(RValue<Pointer<EL>> base, RValue<T> val, RValue<Int4> offsets, RValue<Int4> mask, unsi… in scatter() function
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py435 def scatter(operand, scatter_indices, updates, update_computation, function
/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.cc2279 const XlaComputation& scatter) { in SelectAndScatter()
2295 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
3333 const XlaComputation& scatter) { in SelectAndScatter()
3344 const XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc713 xla::XlaComputation scatter; in ExportXlaOp() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc843 Status IrEmitterUnnested::HandleScatter(HloInstruction* scatter) { in HandleScatter()
889 Thunk* thunk, HloInstruction* scatter, in EmitScatter()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2290 auto scatter = rewriter.create<ScatterOp>( in matchAndRewrite() local
3138 auto scatter = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1081 padding, source, init_value, scatter): argument