| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_index_splitter_test.cc | 68 TEST_F(DynamicIndexSplitterTest, DynamicUpdateSlice) { in TEST_F() argument
91 op::DynamicUpdateSlice(op::Parameter(0), op::Parameter(2), in TEST_F()
|
| D | triangular_solve_expander.cc | 170 DynamicUpdateSlice(neg_identity, pos_one, in InvertDiagonalBlocks()
231 body_out = DynamicUpdateSlice(body_out, update, {zero, j, zero}); in InvertDiagonalBlocks()
|
| D | hlo_cost_analysis_test.cc | 626 TEST_F(HloCostAnalysisTest, DynamicUpdateSlice) { in TEST_F() argument
630 DynamicUpdateSlice(x, ConstantR1<float>(&builder, {1.0}), in TEST_F()
|
| D | hlo_matchers.h | 202 HLO_MATCHER(DynamicUpdateSlice);
|
| D | hlo_parser_test.cc | 624 "DynamicUpdateSlice", in CreateTestCases()
627 ENTRY %DynamicUpdateSlice.v4 (input: s32[1,1,25,1], update: s32[1,1,2,1], start_indices: s32[4]) ->… in CreateTestCases()
641 ENTRY %DynamicUpdateSlice.v4 (input: s32[1,1,25,1], update: s32[1,1,2,1], start_index.0: s32[], sta… in CreateTestCases()
|
| D | hlo_evaluator_typed_visitor.h | 1544 DynamicUpdateSlice<int32>( in HandleDynamicUpdateSlice()
1552 DynamicUpdateSlice<int64>( in HandleDynamicUpdateSlice()
1560 DynamicUpdateSlice<uint32>( in HandleDynamicUpdateSlice()
1568 DynamicUpdateSlice<uint64>( in HandleDynamicUpdateSlice()
2893 StatusOr<Literal> DynamicUpdateSlice( in DynamicUpdateSlice() function
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamic_ops_test.cc | 395 DynamicUpdateSlice(input, update, absl::Span<const XlaOp>({})); in RunR0()
426 DynamicUpdateSlice(input, update, absl::Span<const XlaOp>({starts})); in RunR1()
460 DynamicUpdateSlice(input, update, starts); in RunR2()
500 DynamicUpdateSlice(input, update, starts); in RunR3()
553 DynamicUpdateSlice(input, update, {constant_index, zero, zero}); in RunR3Contiguous()
|
| D | scatter_test.cc | 406 XLA_TEST_F(ScatterTest, DynamicUpdateSlice) { in XLA_TEST_F() argument
|
| D | while_test.cc | 866 auto out1 = DynamicUpdateSlice(input, update, starts); in XLA_TEST_F()
1304 auto out1 = DynamicUpdateSlice(input, update, {zero, zero, zero}); in BM_WhileLoop()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 177 indices = xla::DynamicUpdateSlice(indices, swap_value, in Compile()
180 indices = xla::DynamicUpdateSlice( in Compile()
|
| D | dynamic_slice_ops.cc | 57 xla::XlaOp result = xla::DynamicUpdateSlice( in Compile()
|
| D | tensor_list_ops.cc | 371 OP_REQUIRES_OK(ctx, BuildTensorList(xla::DynamicUpdateSlice(buffer, update, in Compile()
416 BuildTensorList(xla::DynamicUpdateSlice(buffer, update, start_indices), in Compile()
|
| D | stack_ops.cc | 160 b, {xla::DynamicUpdateSlice(ta, update, start_indices), in Compile()
|
| D | tensor_array_ops.cc | 130 return xla::DynamicUpdateSlice(operand, sum, start_indices); in DynamicAddSlice()
231 written = xla::DynamicUpdateSlice(ta, update, start_indices); in Compile()
|
| D | strided_slice_op.cc | 316 lhs = xla::DynamicUpdateSlice(lhs, rhs, slice_begin); in Compile()
|
| D | image_ops.cc | 382 row_iou = xla::DynamicUpdateSlice( in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 65 return DynamicUpdateSlice(x, update, start_ops); in UpdateSlice()
133 return DynamicUpdateSlice(x, update, padded_starts); in DynamicUpdateSliceInMinorDims()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 361 XlaOp DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update,
363 XlaOp DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update,
747 friend XlaOp DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update,
749 friend XlaOp DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update,
1238 XlaOp DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update,
1242 XlaOp DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update,
|
| D | xla_builder.cc | 808 XlaOp XlaBuilder::DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update, in DynamicUpdateSlice() function in xla::XlaBuilder
827 XlaOp XlaBuilder::DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update, in DynamicUpdateSlice() function in xla::XlaBuilder
2890 XlaOp DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update, in DynamicUpdateSlice() function
2892 return operand.builder()->DynamicUpdateSlice(operand, update, start_indices); in DynamicUpdateSlice()
2895 XlaOp DynamicUpdateSlice(const XlaOp& operand, const XlaOp& update, in DynamicUpdateSlice() function
2897 return operand.builder()->DynamicUpdateSlice(operand, update, start_indices); in DynamicUpdateSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 534 LocalOp ComputationBuilder::DynamicUpdateSlice(const LocalOp& operand, in DynamicUpdateSlice() function in xla::swig::ComputationBuilder
537 return xla::DynamicUpdateSlice(operand.op(), update.op(), start_indices.op()); in DynamicUpdateSlice()
|
| D | local_computation_builder.h | 272 LocalOp DynamicUpdateSlice(const LocalOp& operand, const LocalOp& update,
|
| D | local_computation_builder.i | 310 %unignore xla::swig::ComputationBuilder::DynamicUpdateSlice;
|
| D | xla_client.py | 1368 def DynamicUpdateSlice(self, operand, update, start_indices): member in ComputationBuilder
1380 return self._client.DynamicUpdateSlice(operand, update, start_indices)
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 200 // versions of DynamicSlice and DynamicUpdateSlice. Only used for testing.
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1053 ## DynamicUpdateSlice section in Operation Semantics
1056 [`XlaBuilder::DynamicUpdateSlice`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/x…
1058 DynamicUpdateSlice generates a result which is the value of the input array
1065 <b> `DynamicUpdateSlice(operand, update, start_indices)` </b>
1102 DynamicUpdateSlice(a, u, s) produces:
1121 DynamicUpdateSlice(b, u, s) produces:
|