| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_gather_decomposer_test.cc | 56 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
79 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
102 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
127 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
154 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
|
| D | dynamic_index_splitter_test.cc | 69 TEST_F(DynamicIndexSplitterTest, DynamicUpdateSlice) { in TEST_F() argument
92 op::DynamicUpdateSlice(op::Parameter(0), op::Parameter(2), in TEST_F()
|
| D | triangular_solve_expander.cc | 291 DynamicUpdateSlice(neg_identity, pos_one, in InvertDiagonalBlocks()
352 body_out = DynamicUpdateSlice(body_out, update, {zero, j, zero}); in InvertDiagonalBlocks()
|
| D | hlo_cost_analysis_test.cc | 946 TEST_F(HloCostAnalysisTest, DynamicUpdateSlice) { in TEST_F() argument
950 DynamicUpdateSlice(x, ConstantR1<float>(&builder, {1.0}), in TEST_F()
|
| D | hlo_matchers.h | 229 HLO_MATCHER(DynamicUpdateSlice);
|
| D | hlo_evaluator_typed_visitor.h | 1688 DynamicUpdateSlice<int32>(
1696 DynamicUpdateSlice<int64>(
1704 DynamicUpdateSlice<uint32>(
1712 DynamicUpdateSlice<uint64>(
2920 StatusOr<Literal> DynamicUpdateSlice(
|
| D | hlo_parser_test.cc | 725 "DynamicUpdateSlice", in CreateTestCases()
728 ENTRY %DynamicUpdateSlice.v4 (input: s32[1,1,25,1], update: s32[1,1,2,1], start_indices: s32[4]) ->… in CreateTestCases()
742 ENTRY %DynamicUpdateSlice.v4 (input: s32[1,1,25,1], update: s32[1,1,2,1], start_index.0: s32[], sta… in CreateTestCases()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | inplace_ops.cc | 70 current = xla::DynamicUpdateSlice( in Compile()
134 xla::DynamicUpdateSlice(ctx->Input(0), updated, padded_indices); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 166 op::DynamicUpdateSlice( in TEST_F()
188 op::DynamicUpdateSlice( in TEST_F()
311 op::Copy(op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
622 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
2042 AllOf(op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
2043 op::DynamicUpdateSlice( in TEST_F()
2071 AllOf(op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
2072 op::DynamicUpdateSlice( in TEST_F()
3226 AllOf(op::AllReduce(op::DynamicUpdateSlice(_, reshape_l, _, _)), in TEST_F()
3229 AllOf(op::AllReduce(op::DynamicUpdateSlice(_, reshape_r, _, _)), in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unique_op.cc | 138 auto update_input = xla::DynamicUpdateSlice(input, target, {output_index}); in BuildOuterLoopBody()
140 xla::DynamicUpdateSlice(indices, xla::Reshape(output_index, {1}), {i}); in BuildOuterLoopBody()
182 unique_indices = xla::DynamicUpdateSlice( in DataOutputFastPath()
|
| D | random_ops.cc | 188 indices = xla::DynamicUpdateSlice(indices, swap_value, in Compile()
191 indices = xla::DynamicUpdateSlice( in Compile()
|
| D | dynamic_slice_ops.cc | 68 xla::XlaOp result = xla::DynamicUpdateSlice( in Compile()
|
| D | tensor_list_utils.cc | 387 xla::DynamicUpdateSlice(list_part, element_part, start_indices); in ExecuteTensorListPushBack()
403 xla::DynamicUpdateSlice(list_part, update, start_indices); in ExecuteTensorListPushBack()
490 xla::DynamicUpdateSlice(list_part, update, start_indices); in ExecuteTensorListSetItem()
|
| 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 | 468 grad = xla::DynamicUpdateSlice(zero, grad, begins); in CompileAsDynamicUpdateSlice()
698 lhs = xla::DynamicUpdateSlice(lhs, rhs, slice_begin); in Compile()
|
| D | image_ops.cc | 384 row_iou = xla::DynamicUpdateSlice( in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamic_ops_test.cc | 396 DynamicUpdateSlice(input, update, absl::Span<const XlaOp>({})); in RunR0()
427 DynamicUpdateSlice(input, update, absl::Span<const XlaOp>({starts})); in RunR1()
461 DynamicUpdateSlice(input, update, starts); in RunR2()
501 DynamicUpdateSlice(input, update, starts); in RunR3()
554 DynamicUpdateSlice(input, update, {constant_index, zero, zero}); in RunR3Contiguous()
|
| D | scatter_test.cc | 435 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()
1303 auto out1 = DynamicUpdateSlice(input, update, {zero, zero, zero}); in BM_WhileLoop()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 86 return DynamicUpdateSlice(x, update, start_ops); in UpdateSlice()
154 return DynamicUpdateSlice(x, update, padded_starts); in DynamicUpdateSliceInMinorDims()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 157 &DynamicUpdateSlice), in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 500 XlaOp DynamicUpdateSlice(XlaOp operand, XlaOp update,
1101 friend XlaOp DynamicUpdateSlice(XlaOp operand, XlaOp update,
1788 XlaOp DynamicUpdateSlice(XlaOp operand, XlaOp update,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 194 // versions of DynamicSlice and DynamicUpdateSlice. Only used for testing.
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1140 ## DynamicUpdateSlice section in Operation Semantics
1143 [`XlaBuilder::DynamicUpdateSlice`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/x…
1145 DynamicUpdateSlice generates a result which is the value of the input array
1152 <b> `DynamicUpdateSlice(operand, update, start_indices)` </b>
1189 DynamicUpdateSlice(a, u, s) produces:
1208 DynamicUpdateSlice(b, u, s) produces:
|