Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dscatter_expander.cc220 HloInstruction* scatter, HloInstruction* induction_var, in ScatterLoopBody()
328 static int64 ScatterTripCount(HloInstruction* scatter) { in ScatterTripCount()
362 HloInstruction* scatter) { in ExpandInstruction()
Ddfs_hlo_visitor_with_default.h237 Status HandleScatter(HloInstructionPtr scatter) override { in HandleScatter()
Dhlo_instructions.h1365 HloComputation* scatter() const { in scatter() function
Dhlo_cost_analysis.cc1019 Status HloCostAnalysis::HandleScatter(const HloInstruction* scatter) { in HandleScatter()
Dhlo_sharding_util.cc1058 const HloScatterInstruction& scatter) { in IdentityValueAndHloOpcodeForScatterReduceComputation()
Dhlo_verifier.cc1139 Status ShapeVerifier::HandleScatter(HloInstruction* scatter) { in HandleScatter()
Dhlo_instruction.cc1356 HloComputation* scatter) { in CreateSelectAndScatter()
4200 HloComputation* HloInstruction::scatter() const { in scatter() function in xla::HloInstruction
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_select_and_scatter_op.cc93 XlaCompiler::CompilationResult scatter; in Compile() local
Dpooling_ops.cc357 auto scatter = CreateScalarAddComputation(element_type, ctx->builder()); in Compile() local
/external/libdrm/
Dxf86drmHash.c85 static unsigned long scatter[256]; in HashHash() local
/external/tensorflow/tensorflow/python/ops/
Dtensor_array_ops.py331 def scatter(self, indices, value, name=None): member in _GraphTensorArray
599 def scatter(self, indices, value, name=None): member in _GraphTensorArrayV2
891 def scatter(self, indices, value, name=None): member in _EagerTensorArray
1224 def scatter(self, indices, value, name=None): member in TensorArray
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_array_ops_decomposition.cc135 } else if (auto scatter = in GetTensorArrayElementShape() local
444 TF::TensorArrayScatterV3Op scatter, in HandleTensorArrayScatterV3Op()
888 } else if (auto scatter = llvm::dyn_cast<TF::TensorArrayScatterV3Op>(&op)) { in DecomposeTensorArrayOps() local
Dtensor_list_ops_decomposition.cc788 TF::TensorListScatterIntoExistingListOp scatter, in HandleTensorListScatterIntoExistingListOp()
868 } else if (auto scatter = in DecomposeTensorListOpsInternal() local
Dshape_inference.cc202 if (auto scatter = llvm::dyn_cast<TensorListScatterIntoExistingListOp>( in CanInferTensorListElementType() local
/external/mesa3d/src/glx/
Dglxhash.c132 static unsigned long scatter[256]; in HashHash() local
/external/swiftshader/src/Reactor/
DEmulatedIntrinsics.cpp84 void scatter(RValue<Pointer<EL>> base, RValue<T> val, RValue<Int4> offsets, RValue<Int4> mask, unsi… in scatter() function
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc508 auto scatter = Cast<HloScatterInstruction>(hlo); in HandleScatter() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc431 auto scatter = Cast<HloScatterInstruction>(instruction); in ImportInstructionImpl() local
Dmlir_hlo_to_hlo.cc964 xla::XlaComputation scatter; in ExportXlaOp() local
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py493 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-10.0/llvm/lib/Transforms/Scalar/
DScalarizer.cpp345 Scatterer ScalarizerVisitor::scatter(Instruction *Point, Value *V) { in scatter() function in ScalarizerVisitor
/external/llvm-project/llvm/lib/Transforms/Scalar/
DScalarizer.cpp355 Scatterer ScalarizerVisitor::scatter(Instruction *Point, Value *V) { in scatter() function in ScalarizerVisitor
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3031 const XlaComputation& scatter) { in SelectAndScatter()
3076 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterInternal()
3106 XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
4706 const XlaComputation& scatter) { in SelectAndScatter()
4717 const XlaOp init_value, const XlaComputation& scatter) { in SelectAndScatterWithGeneralPadding()
/external/llvm-project/mlir/lib/Conversion/VectorToLLVM/
DConvertVectorToLLVM.cpp454 auto scatter = cast<vector::ScatterOp>(op); in matchAndRewrite() local

12