| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 177 op::Copy(op::DynamicSlice( in TEST_F()
181 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
203 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
225 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
320 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())); in TEST_F()
323 op::DynamicSlice(op::GetTupleElement(op::Parameter()), offset, in TEST_F()
326 op::DynamicSlice(op::GetTupleElement(op::Parameter()), offset, in TEST_F()
350 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
485 op::DynamicSlice(op::Pad(op::Constant(), op::Constant()), in TEST_F()
514 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 | reduce_scatter_decomposer_test.cc | 80 slice_index = op::Reshape(op::DynamicSlice(op::Constant(), id_matcher)); in RunPass()
96 EXPECT_THAT(root, op::DynamicSlice(ds_operands)); in RunPass()
|
| D | all_gather_decomposer_test.cc | 124 op::Reshape(op::DynamicSlice(op::Constant(), op::ReplicaId()))); in TEST_F()
151 op::Reshape(op::DynamicSlice(op::Constant(), global_id))); in TEST_F()
|
| D | hlo_matchers.h | 267 HLO_MATCHER(DynamicSlice);
331 HLO_MATCHER_VECTOR_OPERANDS(DynamicSlice);
|
| D | hlo_evaluator_typed_visitor.h | 1438 DynamicSlice<int32_t>(
1446 DynamicSlice<int64_t>(
1454 DynamicSlice<uint32_t>(
1462 DynamicSlice<uint64_t>(
2345 StatusOr<Literal> DynamicSlice(
|
| D | instruction_fusion_test.cc | 544 EXPECT_THAT(root, op::Fusion(op::Parameter(), op::DynamicSlice())); in TEST_F()
|
| D | hlo_cost_analysis_test.cc | 1095 TEST_F(HloCostAnalysisTest, DynamicSlice) { in TEST_F() argument
1099 DynamicSlice(x, absl::Span<const XlaOp>({ConstantR0<int32_t>(&builder, 1)}), in TEST_F()
|
| D | triangular_solve_expander.cc | 331 DynamicSlice(body_input, {zero, j, zero}, in InvertDiagonalBlocks()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unique_op.cc | 113 auto data = xla::DynamicSlice(data_stack, {counter}, {1}); in RollingSelectR1()
117 auto mask = xla::DynamicSlice(mask_stack, {counter}, {1}); in RollingSelectR1()
126 auto accum_minus = xla::DynamicSlice(accum_stack, {counter_minus}, {1}); in RollingSelectR1()
|
| D | slice_op.cc | 155 ctx->SetOutput(0, xla::DynamicSlice(input, begin_indices, size)); in Compile()
175 auto sliced = xla::DynamicSlice(padded_input, begin_indices, in Compile()
|
| D | roll_op.cc | 87 xla::DynamicSlice(concat, start_indices, input_shape.dim_sizes()); in Compile()
|
| D | stack_ops.cc | 213 xla::XlaOp read = xla::DynamicSlice(ta, start_indices, slice_shape); in Compile()
|
| D | dynamic_partition_op.cc | 104 xla::DynamicSlice(padded_data, {partition_start[i]}, {input_count}); in DynamicPartition1D()
|
| 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 | 447 xla::XlaOp read = xla::DynamicSlice(list_part, start_indices, slice_shape); in ExecuteTensorListPopBack()
528 xla::XlaOp read = xla::DynamicSlice(list_part, start_indices, slice_shape); in ExecuteTensorListGetItem()
|
| D | image_ops.cc | 381 auto active_elem = xla::DynamicSlice(included_iou, row_idx_vector, {1}); in operator ()()
388 auto row_iou = xla::DynamicSlice(iou_mask, {row_idx, zero}, {1, num_boxes}); in operator ()()
|
| /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 | dot_operation_test.cc | 966 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
994 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1024 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1052 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
1085 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1118 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F()
1143 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F()
1168 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()
799 DynamicSlice(input, start_indices, {1, 1, 1, 1}); in BM_DynamicSlice()
|
| 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/xla_extension/ |
| D | ops.pyi | 220 def DynamicSlice(
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 226 absl::Span<const int64_t>)>(&DynamicSlice), in BuildOpsSubmodule()
|