| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 142 op::Copy(op::DynamicSlice( in TEST_F()
146 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
168 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
190 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
285 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())); in TEST_F()
288 op::DynamicSlice(op::GetTupleElement(op::Parameter()), offset, in TEST_F()
291 op::DynamicSlice(op::GetTupleElement(op::Parameter()), offset, in TEST_F()
315 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
450 op::DynamicSlice(op::Pad(op::Constant(), op::Constant()), in TEST_F()
479 op::DynamicSlice(op::Pad(op::Constant(), op::Constant()), in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_index_splitter_test.cc | 32 TEST_F(DynamicIndexSplitterTest, DynamicSlice) { in TEST_F() argument
54 op::DynamicSlice(op::Parameter(0), in TEST_F()
131 op::DynamicSlice(op::Parameter(0), op::Parameter(1), in TEST_F()
|
| D | all_gather_decomposer_test.cc | 125 op::Reshape(op::DynamicSlice(op::Constant(), op::ReplicaId()))); in TEST_F()
152 op::Reshape(op::DynamicSlice(op::Constant(), global_id))); in TEST_F()
|
| D | hlo_cost_analysis_test.cc | 925 TEST_F(HloCostAnalysisTest, DynamicSlice) { in TEST_F() argument
929 DynamicSlice(x, absl::Span<const XlaOp>({ConstantR0<int32>(&builder, 1)}), in TEST_F()
|
| D | hlo_matchers.h | 228 HLO_MATCHER(DynamicSlice);
|
| D | hlo_evaluator_typed_visitor.h | 1623 DynamicSlice<int32>(
1631 DynamicSlice<int64>(
1639 DynamicSlice<uint32>(
1647 DynamicSlice<uint64>(
2885 StatusOr<Literal> DynamicSlice(
|
| D | hlo_parser_test.cc | 696 "DynamicSlice", in CreateTestCases()
699 ENTRY %DynamicSlice.v5 (original_parameter: s32[2,2,258], start_index: s32[1]) -> s32[2,2,258] { in CreateTestCases()
714 ENTRY %DynamicSlice.v5 (original_parameter: s32[2,2,258], start_index: s32[]) -> s32[2,2,258] { in CreateTestCases()
|
| D | triangular_solve_expander.cc | 338 DynamicSlice(body_input, {zero, j, zero}, in InvertDiagonalBlocks()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 180 xla::DynamicSlice(indices, absl::Span<const xla::XlaOp>({i}), {1}); in Compile()
183 xla::DynamicSlice(swaps, absl::Span<const xla::XlaOp>({i}), {1}), {}); in Compile()
185 auto swap_value = xla::DynamicSlice( in Compile()
|
| D | slice_op.cc | 146 xla::DynamicSlice(ctx->Input(0), scalar_indices, size)); in Compile()
165 auto sliced = xla::DynamicSlice(padded_input, scalar_indices, in Compile()
|
| D | roll_op.cc | 87 xla::DynamicSlice(concat, start_indices, input_shape.dim_sizes()); in Compile()
|
| D | dynamic_slice_ops.cc | 113 xla::XlaOp result = xla::DynamicSlice( in Compile()
|
| D | unique_op.cc | 92 auto input_elem = xla::DynamicSlice(input, {j}, {1}); in BuildInnerLoopBody()
127 auto target = xla::DynamicSlice(input, {i}, {1}); in BuildOuterLoopBody()
|
| 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 | tensor_list_utils.cc | 445 xla::XlaOp read = xla::DynamicSlice(list_part, start_indices, slice_shape); in ExecuteTensorListPopBack()
525 xla::XlaOp read = xla::DynamicSlice(list_part, start_indices, slice_shape); in ExecuteTensorListGetItem()
|
| D | image_ops.cc | 374 auto active_elem = xla::DynamicSlice(included_iou, row_idx_vector, {1}); in operator ()()
381 auto row_iou = xla::DynamicSlice(iou_mask, {row_idx, zero}, {1, num_boxes}); in operator ()()
|
| D | strided_slice_op.cc | 186 xla::DynamicSlice(slice, start_indices, processing_shape.dim_sizes()); in EmitDynamicSlice()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | get_item_op.cc | 65 auto slice = xla::DynamicSlice(ctx->Input(0), operands, slice_sizes); in Compile()
|
| D | inplace_ops.cc | 131 auto prev = xla::DynamicSlice(ctx->Input(0), padded_indices, sizes); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamic_ops_test.cc | 143 DynamicSlice(input, absl::Span<const XlaOp>({starts}), slice_sizes); in RunR1()
173 DynamicSlice(input, starts, slice_sizes); in RunR2()
207 DynamicSlice(input, starts, slice_sizes); in RunR3()
796 DynamicSlice(input, start_indices, {1, 1, 1, 1}); in BM_DynamicSlice()
|
| D | dot_operation_test.cc | 965 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
993 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1023 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1051 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
1084 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1117 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1142 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
1167 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
|
| D | gather_operation_test.cc | 223 XLA_TEST_F(GatherOperationTest, DynamicSlice) { in XLA_TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 32 XlaOp sliced_input = DynamicSlice(input, base_indices, window_sizes); in DynamicStridedSlice()
145 return DynamicSlice(x, padded_starts, padded_sizes); in DynamicSliceInMinorDims()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 153 absl::Span<const int64>)>(&DynamicSlice), in BuildOpsSubmodule()
|