| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamic_ops_test.cc | 118 const std::vector<IndexT> slice_starts, in RunR1() argument
139 slice_starts[0], 0, "slice_starts", &builder, &starts); in RunR1()
149 const std::vector<IndexT> slice_starts, in RunR2() argument
167 slice_starts[i], i, "slice_starts", &builder, &starts[i]); in RunR2()
184 const std::vector<IndexT> slice_starts, in RunR3() argument
202 slice_starts[i], i, "slice_starts", &builder, &starts[i]); in RunR3()
377 const std::vector<IndexT> slice_starts, int expected_value_int) { in RunR0() argument
403 const std::vector<IndexT> slice_starts, in RunR1() argument
422 slice_starts[0], 0, "slice_starts", &builder, &starts); in RunR1()
434 const std::vector<IndexT> slice_starts, in RunR2() argument
[all …]
|
| D | slice_test.cc | 379 std::array<int64, 2> slice_starts; member
399 Slice(a, spec.slice_starts, spec.slice_limits, spec.slice_strides); in XLA_TEST_P()
404 input, spec.slice_starts, spec.slice_limits, spec.slice_strides); in XLA_TEST_P()
464 std::array<int64, 4> slice_starts; member
473 "__starts_", absl::StrJoin(spec.slice_starts, "x"), in R4SpecToString()
486 values, spec.slice_starts, spec.slice_limits, spec.slice_strides); in Run()
493 Slice(parameter, spec.slice_starts, spec.slice_limits, spec.slice_strides); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_index_splitter_test.cc | 63 EXPECT_EQ(slice->slice_starts(0), i); in TEST_F()
101 EXPECT_EQ(slice->slice_starts(0), i); in TEST_F()
|
| D | algebraic_simplifier.cc | 690 operands[j]->slice_starts(concatenate_dimension) == slice_end) { in HandleConcatenate()
696 for (int64 k = 0; k < operands[j]->slice_starts().size(); ++k) { in HandleConcatenate()
700 if (operands[i]->slice_starts(k) != operands[j]->slice_starts(k)) { in HandleConcatenate()
715 slice_end - operands[i]->slice_starts(concatenate_dimension)); in HandleConcatenate()
721 /*start_indices=*/operands[i]->slice_starts(), in HandleConcatenate()
2693 int64 start = slice->slice_starts(i); in TrySimplifyScalarSlice()
2741 if (next_operand_start > slice->slice_starts(0)) { in TrySimplifyScalarSlice()
2757 {slice->slice_starts(0) - operand_start}, in TrySimplifyScalarSlice()
2758 {slice->slice_starts(0) - operand_start + 1}, in TrySimplifyScalarSlice()
2781 if (slice->slice_starts(i) != 0 || in TryToReorderSliceAndReshape()
[all …]
|
| D | hlo_instruction.cc | 289 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
292 slice_starts.push_back(slice_dimensions.start()); in CreateFromProto()
296 instruction = CreateSlice(shape, operands(0), slice_starts, slice_limits, in CreateFromProto()
3247 int64 HloInstruction::slice_starts(int64 dimension) const { in slice_starts() function in xla::HloInstruction
3248 return Cast<HloSliceInstruction>(this)->slice_starts(dimension); in slice_starts()
3251 const std::vector<int64>& HloInstruction::slice_starts() const { in slice_starts() function in xla::HloInstruction
3252 return Cast<HloSliceInstruction>(this)->slice_starts(); in slice_starts()
|
| D | hlo_instruction.h | 1410 int64 slice_starts(int64 dimension) const;
1411 const std::vector<int64>& slice_starts() const;
|
| D | dynamic_dimension_inference.cc | 398 if (hlo->slice_starts(dimension) != 0 || in HandleSlice()
|
| D | hlo_instructions.h | 618 int64 slice_starts(int64 dimension) const { return slice_starts_[dimension]; } in slice_starts() function
619 const std::vector<int64>& slice_starts() const { return slice_starts_; } in slice_starts() function
|
| D | hlo_verifier.cc | 571 slice->operand(0)->shape(), slice->slice_starts(), in HandleSlice()
|
| D | algebraic_simplifier_test.cc | 1875 EXPECT_EQ(computation->root_instruction()->operand(3)->slice_starts(1), 40); in TEST_F()
2715 EXPECT_EQ(computation->root_instruction()->slice_starts(0), 3); in TEST_F()
2716 EXPECT_EQ(computation->root_instruction()->slice_starts(1), 5); in TEST_F()
4132 EXPECT_EQ(root->slice_starts(0), 1); in TEST_F()
|
| D | hlo_evaluator_typed_visitor.h | 2419 operand->shape(), slice->slice_starts(), in HandleSlice()
2432 slice->slice_starts(i) + out_index[i] * slice->slice_strides(i); in HandleSlice()
|
| D | elemental_ir_emitter.cc | 2370 /*starts=*/hlo->slice_starts(), in MakeElementGenerator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1989 ? slice->slice_limits(memcpy_dim) - slice->slice_starts(memcpy_dim) in HandleSlice()
2026 /*operand_shape=*/operand->shape(), /*starts=*/slice->slice_starts(), in HandleSlice()
|