| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_index_splitter_test.cc | 32 TEST_F(DynamicIndexSplitterTest, DynamicSlice) { in TEST_F() argument
53 op::DynamicSlice(op::Parameter(0), in TEST_F()
129 op::DynamicSlice(op::Parameter(0), op::Parameter(1), in TEST_F()
|
| D | hlo_cost_analysis_test.cc | 610 TEST_F(HloCostAnalysisTest, DynamicSlice) { in TEST_F() argument
614 DynamicSlice(x, absl::Span<const XlaOp>({ConstantR0<int32>(&builder, 1)}), in TEST_F()
|
| D | hlo_matchers.h | 201 HLO_MATCHER(DynamicSlice);
|
| D | triangular_solve_expander.cc | 217 DynamicSlice(body_input, {zero, j, zero}, in InvertDiagonalBlocks()
|
| D | hlo_parser_test.cc | 595 "DynamicSlice", in CreateTestCases()
598 ENTRY %DynamicSlice.v5 (original_parameter: s32[2,2,258], start_index: s32[1]) -> s32[2,2,258] { in CreateTestCases()
613 ENTRY %DynamicSlice.v5 (original_parameter: s32[2,2,258], start_index: s32[]) -> s32[2,2,258] { in CreateTestCases()
|
| D | hlo_evaluator_typed_visitor.h | 1479 DynamicSlice<int32>( in HandleDynamicSlice()
1487 DynamicSlice<int64>( in HandleDynamicSlice()
1495 DynamicSlice<uint32>( in HandleDynamicSlice()
1503 DynamicSlice<uint64>( in HandleDynamicSlice()
2858 StatusOr<Literal> DynamicSlice( in DynamicSlice() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 169 xla::DynamicSlice(indices, absl::Span<const xla::XlaOp>({i}), {1}); in Compile()
172 xla::DynamicSlice(swaps, absl::Span<const xla::XlaOp>({i}), {1}), {}); in Compile()
174 auto swap_value = xla::DynamicSlice( in Compile()
|
| D | dynamic_slice_ops.cc | 99 xla::XlaOp result = xla::DynamicSlice( in Compile()
|
| D | slice_op.cc | 110 ctx->SetOutput(0, xla::DynamicSlice(ctx->Input(0), ctx->Input(1), size)); in Compile()
|
| D | tensor_list_ops.cc | 230 xla::XlaOp read = xla::DynamicSlice(buffer, start_indices, slice_shape); in Compile()
456 xla::XlaOp read = xla::DynamicSlice(ta, start_indices, slice_shape); in Compile()
|
| D | stack_ops.cc | 213 xla::XlaOp read = xla::DynamicSlice(ta, start_indices, slice_shape); in Compile()
|
| D | tensor_array_ops.cc | 127 xla::XlaOp current = xla::DynamicSlice(operand, start_indices, update_dims); in DynamicAddSlice()
273 xla::XlaOp read = xla::DynamicSlice(ta, start_indices, slice_shape); in Compile()
|
| D | image_ops.cc | 372 auto active_elem = xla::DynamicSlice(included_iou, row_idx_vector, {1}); in operator ()()
379 auto row_iou = xla::DynamicSlice(iou_mask, {row_idx, zero}, {1, num_boxes}); in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 925 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
953 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
983 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1011 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
1044 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1077 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1102 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
1127 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
|
| D | dynamic_ops_test.cc | 142 DynamicSlice(input, absl::Span<const XlaOp>({starts}), slice_sizes); in RunR1()
172 DynamicSlice(input, starts, slice_sizes); in RunR2()
206 DynamicSlice(input, starts, slice_sizes); in RunR3()
779 DynamicSlice(input, start_indices, {1, 1, 1, 1}); in BM_DynamicSlice()
|
| D | gather_operation_test.cc | 199 XLA_TEST_F(GatherOperationTest, DynamicSlice) { in XLA_TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 354 XlaOp DynamicSlice(const XlaOp& operand, const XlaOp& start_indices,
356 XlaOp DynamicSlice(const XlaOp& operand,
741 friend XlaOp DynamicSlice(const XlaOp& operand, const XlaOp& start_indices,
743 friend XlaOp DynamicSlice(const XlaOp& operand,
1214 XlaOp DynamicSlice(const XlaOp& operand, absl::Span<const XlaOp> start_indices,
1218 XlaOp DynamicSlice(const XlaOp& operand, const XlaOp& start_indices,
|
| D | xla_builder.cc | 758 XlaOp XlaBuilder::DynamicSlice(const XlaOp& operand, const XlaOp& start_indices, in DynamicSlice() function in xla::XlaBuilder
780 XlaOp XlaBuilder::DynamicSlice(const XlaOp& operand, in DynamicSlice() function in xla::XlaBuilder
2881 XlaOp DynamicSlice(const XlaOp& operand, const XlaOp& start_indices, in DynamicSlice() function
2883 return operand.builder()->DynamicSlice(operand, start_indices, slice_sizes); in DynamicSlice()
2885 XlaOp DynamicSlice(const XlaOp& operand, absl::Span<const XlaOp> start_indices, in DynamicSlice() function
2887 return operand.builder()->DynamicSlice(operand, start_indices, slice_sizes); in DynamicSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 124 return DynamicSlice(x, padded_starts, padded_sizes); in DynamicSliceInMinorDims()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 528 LocalOp ComputationBuilder::DynamicSlice(const LocalOp& operand, in DynamicSlice() function in xla::swig::ComputationBuilder
531 return xla::DynamicSlice(operand.op(), start_indices.op(), slice_sizes); in DynamicSlice()
|
| D | local_computation_builder.h | 269 LocalOp DynamicSlice(const LocalOp& operand, const LocalOp& start_indices,
|
| D | local_computation_builder.i | 309 %unignore xla::swig::ComputationBuilder::DynamicSlice;
|
| D | xla_client.py | 1353 def DynamicSlice(self, operand, start_indices, slice_sizes): member in ComputationBuilder
1366 return self._client.DynamicSlice(operand, start_indices, slice_sizes)
|
| /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 | 989 ## DynamicSlice section in Operation Semantics
992 [`XlaBuilder::DynamicSlice`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_bui…
994 DynamicSlice extracts a sub-array from the input array at dynamic
1000 <b> `DynamicSlice(operand, start_indices, size_indices)` </b>
1035 DynamicSlice(a, s, {2}) produces:
1049 DynamicSlice(b, s, {2, 2}) produces:
|