Home
last modified time | relevance | path

Searched refs:Scatter (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TensorArrayScatterV3.pbtxt36 summary: "Scatter the data from the input value into specific TensorArray elements."
Dapi_def_TensorScatterUpdate.pbtxt28 summary: "Scatter `updates` into an existing tensor according to `indices`."
Dapi_def_ScatterNd.pbtxt28 summary: "Scatter `updates` into a new tensor according to `indices`."
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc202 return xla::Scatter(buffer, indices, new_updates, combiner_computation, in XlaScatter()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc667 TEST_F(HloCostAnalysisTest, Scatter) { in TEST_F() argument
682 Scatter(operand, indices, values, add_, dim_numbers); in TEST_F()
Dhlo_matchers.h232 HLO_MATCHER(Scatter);
Dhlo_parser_test.cc868 ENTRY %Scatter (input_tensor: f32[50,49,48,47,46], scatter_indices: s64[10,9,8,7,5], updates: f32[1… in CreateTestCases()
Dpattern_matcher.h2221 XLA_TERNOP_PATTERN(Scatter);
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.h717 Status Scatter(OpKernelContext* c, const Tensor& value, const Tensor& indices, in Scatter() function
786 Scatter<Device, T>(c, input_tensor, indices, output_list)); in Compute()
853 Scatter<Device, T>(c, input_tensor, indices, &output_list)); in Compute()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc762 LocalOp ComputationBuilder::Scatter( in Scatter() function in xla::swig::ComputationBuilder
766 return xla::Scatter(input.op(), scatter_indices.op(), updates.op(), in Scatter()
Dlocal_computation_builder.h377 LocalOp Scatter(const LocalOp& input, const LocalOp& scatter_indices,
Dlocal_computation_builder.i402 %unignore xla::swig::ComputationBuilder::Scatter;
Dxla_client.py1779 def Scatter(self, a, scatter_indices, updates, update_computation, member in ComputationBuilder
1782 return self._client.Scatter(
Dxla_client_test.py1690 c.Scatter(c.Constant(a), c.Constant(scatter_indices), c.Constant(updates),
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc607 op::Scatter(op::Add(), op::Add(), op::Add())); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc187 return xla::Scatter(grad_data, indices, updates, in ScatterToGradData()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h547 XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices,
971 friend XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices,
1820 XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices,
Dxla_builder.cc1864 XlaOp XlaBuilder::Scatter(const XlaOp& input, const XlaOp& scatter_indices, in Scatter() function in xla::XlaBuilder
3431 XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices, in Scatter() function
3434 return input.builder()->Scatter(input, scatter_indices, updates, in Scatter()
/external/llvm/docs/
DVectorizers.rst260 Scatter / Gather
/external/swiftshader/third_party/llvm-7.0/llvm/docs/
DVectorizers.rst262 Scatter / Gather
/external/llvm/lib/Target/SystemZ/
DSystemZInstrVector.td215 // Scatter element.
/external/swiftshader/third_party/llvm-7.0/llvm/test/CodeGen/X86/
Davx512-gather-scatter-intrin.ll78 ;; Integer Gather/Scatter
/external/llvm/test/CodeGen/X86/
Davx512-gather-scatter-intrin.ll74 ;; Integer Gather/Scatter
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2106 ## Scatter section in Operation Semantics
2113 [`XlaBuilder::Scatter`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builder.…
/external/llvm/lib/CodeGen/SelectionDAG/
DSelectionDAGBuilder.cpp3809 SDValue Scatter = DAG.getMaskedScatter(DAG.getVTList(MVT::Other), VT, sdl, in visitMaskedScatter() local
3811 DAG.setRoot(Scatter); in visitMaskedScatter()
3812 setValue(&I, Scatter); in visitMaskedScatter()

12