| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | scatter_test.cc | 28 Literal* scatter_indices, Literal* updates) { in RunTest()
65 Literal updates = in XLA_TEST_F() local
97 Literal updates = in XLA_TEST_F() local
126 Literal updates = in XLA_TEST_F() local
192 Literal updates = in XLA_TEST_F() local
223 Literal updates = in XLA_TEST_F() local
254 Literal updates = in XLA_TEST_F() local
284 Literal updates = in XLA_TEST_F() local
314 Literal updates = in XLA_TEST_F() local
344 Literal updates = in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 383 def scatter_update(ref, indices, updates, use_locking=True, name=None): argument
437 def scatter_nd_update(ref, indices, updates, use_locking=True, name=None): argument
499 def scatter_add(ref, indices, updates, use_locking=False, name=None): argument
551 def scatter_nd_add(ref, indices, updates, use_locking=False, name=None): argument
614 def scatter_sub(ref, indices, updates, use_locking=False, name=None): argument
668 def scatter_nd_sub(ref, indices, updates, use_locking=False, name=None): argument
732 def scatter_mul(ref, indices, updates, use_locking=False, name=None): argument
784 def scatter_div(ref, indices, updates, use_locking=False, name=None): argument
836 def scatter_max(ref, indices, updates, use_locking=False, name=None): argument
891 def scatter_min(ref, indices, updates, use_locking=False, name=None): argument
[all …]
|
| D | resource_variable_ops.py | 1185 def scatter_nd_sub(self, indices, updates, name=None): argument
1237 def scatter_nd_add(self, indices, updates, name=None): argument
1289 def scatter_nd_update(self, indices, updates, name=None): argument
1341 def scatter_nd_max(self, indices, updates, name=None): argument
1377 def scatter_nd_min(self, indices, updates, name=None): argument
2226 def scatter_nd_sub(self, indices, updates, name=None): argument
2230 def scatter_nd_add(self, indices, updates, name=None): argument
2234 def scatter_nd_update(self, indices, updates, name=None): argument
2239 def scatter_nd_max(self, indices, updates, name=None): argument
2243 def scatter_nd_min(self, indices, updates, name=None): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | scatter_nd.cc | 51 const RuntimeShape& updates, in CheckShapes()
79 const TfLiteTensor* updates; in Prepare() local
130 TfLiteStatus ScatterNd(const TfLiteTensor* indices, const TfLiteTensor* updates, in ScatterNd()
140 const TfLiteTensor* updates, in EvalScatterNd()
186 const TfLiteTensor* updates; in Eval() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 59 def _NumpyScatterNd(ref, indices, updates, op): argument
76 def _NumpyUpdate(ref, indices, updates): argument
80 def _NumpyAdd(ref, indices, updates): argument
84 def _NumpySub(ref, indices, updates): argument
88 def _NumpyMul(ref, indices, updates): argument
92 def _NumpyDiv(ref, indices, updates): argument
96 def _NumpyMin(ref, indices, updates): argument
100 def _NumpyMax(ref, indices, updates): argument
490 def scatter_nd(self, indices, updates, shape, input_=None): argument
785 def scatter_nd(self, indices, updates, shape, input_=None): argument
|
| D | scatter_ops_test.py | 32 def _NumpyAdd(ref, indices, updates): argument
44 def _NumpySub(ref, indices, updates): argument
54 def _NumpyMul(ref, indices, updates): argument
64 def _NumpyDiv(ref, indices, updates): argument
74 def _NumpyMin(ref, indices, updates): argument
84 def _NumpyMax(ref, indices, updates): argument
94 def _NumpyUpdate(ref, indices, updates): argument
|
| D | batch_scatter_ops_test.py | 31 def _NumpyUpdate(ref, indices, updates): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/ |
| D | scatter_nd_ops_test.py | 45 def _NumpyScatterNd(ref, indices, updates, op): argument
62 def _NumpyMin(ref, indices, updates): argument
66 def _NumpyMax(ref, indices, updates): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op.cc | 34 static bool ValidShapes(const Tensor& params, const Tensor& updates, in ValidShapes()
52 const Tensor& indices, const Tensor& updates) { in DoValidationChecking()
104 const Tensor& updates = c->input(2); in DoCompute() local
|
| D | scatter_nd_op.cc | 72 const Tensor& updates = c->input(1); in Compute() local
149 const Tensor& updates = c->input(2); in Compute() local
277 const Tensor& updates = c->input(2); in DoCompute() local
778 const Tensor& indices, const Tensor& updates, in PrepareAndValidateInputs()
869 const Tensor& updates, const TensorShape& shape, in DoScatterNdOnCpu()
945 const Tensor& updates, const TensorShape& shape, Tensor* out, in DoScatterNd()
|
| D | resource_variable_ops.cc | 921 bool ValidateInput(const Tensor& updates) { in ValidateInput()
930 bool ValidateInput<Variant>(const Tensor& updates) { in ValidateInput()
966 const Tensor& updates, Index num_indices) { in DoScatterOnCpu()
1000 const Tensor& updates, Index num_indices) { in DoScatter()
1068 const Tensor& updates = c->input(2); in DoCompute() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scatter_nd_op_test.py | 46 def _NumpyScatterNd(ref, indices, updates, op): argument
63 def _NumpyUpdate(indices, updates, shape): argument
118 def _runScatterNd(self, indices, updates, shape): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 98 HloInstruction* updates, absl::Span<const int64_t> update_window_dims) { in PermuteScatterAndWindowDims()
118 const Shape& scatter_indices_shape, HloInstruction* updates, in AdjustScatterDims()
256 auto updates = loop_state.last(operands.size()); in ScatterLoopBody() local
|
| D | dynamic_update_slice_test.cc | 168 Literal updates = in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | journal_test.cc | 92 std::vector<Update> updates = {MakeCreateIterationUpdate(), in TEST() local
106 std::vector<Update> updates = {MakeCreateIterationUpdate(), in TEST() local
|
| /external/nullaway/nullaway/src/main/java/com/uber/nullaway/dataflow/ |
| D | AccessPathNullnessPropagation.java | 509 ReadableUpdates updates = new ReadableUpdates(); in visitAssignment() local
561 ReadableUpdates updates) { in handleEnhancedForOverKeySet()
648 Nullness value, TransferInput<Nullness, NullnessStore> input, ReadableUpdates updates) { in updateRegularStore()
657 private void setNonnullIfAnalyzeable(Updates updates, Node node) { in setNonnullIfAnalyzeable()
694 ReadableUpdates updates = new ReadableUpdates(); in visitVariableDeclaration() local
709 ReadableUpdates updates = new ReadableUpdates(); in visitFieldAccess() local
722 AccessPathNullnessPropagation.ReadableUpdates updates, Node receiver, Symbol symbol) { in setReceiverNonnull()
737 ReadableUpdates updates = new ReadableUpdates(); in visitArrayAccess() local
851 ReadableUpdates updates = new ReadableUpdates(); in visitAssertionError() local
1021 private static ResultingStore updateStore(NullnessStore oldStore, ReadableUpdates... updates) { in updateStore()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values_v2.py | 203 def scatter_nd_sub(self, indices, updates, name=None): argument
207 def scatter_nd_add(self, indices, updates, name=None): argument
211 def scatter_nd_update(self, indices, updates, name=None): argument
|
| /external/libchrome/base/files/ |
| D | file_path_watcher_kqueue.cc | 211 EventVector updates(valid); in UpdateWatches() local
307 EventVector updates(events_.size()); in OnKQueueReadable() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 68 const PartitionedHlo& updates, const PartitionedHlo& indices, in ComputeUpdateShardingFromIndices()
775 PartitionedHlo& indices, absl::Span<PartitionedHlo> updates, in PartitionScatterPassthroughOperand()
814 PartitionedHlo& indices, absl::Span<PartitionedHlo> updates, in PartitionScatterIndexPassthroughPartition()
953 PartitionedHlo& indices, absl::Span<PartitionedHlo> updates, in PartitionScatterTrivialIndexedOperandDimension()
1009 PartitionedHlo& indices, absl::Span<PartitionedHlo> updates, in PartitionScatter()
1046 absl::InlinedVector<PartitionedHlo, 1> operands, updates; in HandleScatter() local
|
| /external/cronet/base/files/ |
| D | file_path_watcher_kqueue.cc | 235 EventVector updates(valid); in UpdateWatches() local
341 EventVector updates(events_.size()); in OnKQueueReadable() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | scatter_nd_op.cc | 119 auto updates = context->Input(1); in Compile() local
174 auto updates = context->Input(2); in CompileTensorScatter() local
|
| D | bincount_op.cc | 56 xla::XlaOp idx, updates, output; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | scatter_nd_fuzz.cc | 29 auto updates = in BuildGraph() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | scatter.cc | 35 const xla::XlaOp& buffer, const xla::XlaOp& updates, in XlaScatter()
|
| /external/nullaway/nullaway/src/main/java/com/uber/nullaway/handlers/ |
| D | ApacheThriftIsSetHandler.java | 96 AccessPathNullnessPropagation.Updates updates, in updateNonNullAPsForElement()
|