| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 380 std::array<int64, 2> slice_limits; 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()
465 std::array<int64, 4> slice_limits; member
474 "__limits_", absl::StrJoin(spec.slice_limits, "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 | 64 EXPECT_EQ(slice->slice_limits(0), i + 1); in TEST_F()
102 EXPECT_EQ(slice->slice_limits(0), i + 1); in TEST_F()
|
| D | hlo_constant_folding_test.cc | 154 const int64 slice_limits[] = {10, 8, 6, 5, 9}; in TEST_F() local
163 shape, literal_instruction, slice_start, slice_limits, slice_strides)); in TEST_F()
|
| D | hlo_instruction.cc | 289 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
293 slice_limits.push_back(slice_dimensions.limit()); in CreateFromProto()
296 instruction = CreateSlice(shape, operands(0), slice_starts, slice_limits, in CreateFromProto()
3255 int64 HloInstruction::slice_limits(int64 dimension) const { in slice_limits() function in xla::HloInstruction
3256 return Cast<HloSliceInstruction>(this)->slice_limits(dimension); in slice_limits()
3259 const std::vector<int64>& HloInstruction::slice_limits() const { in slice_limits() function in xla::HloInstruction
3260 return Cast<HloSliceInstruction>(this)->slice_limits(); in slice_limits()
|
| D | hlo_instruction.h | 1414 int64 slice_limits(int64 dimension) const;
1415 const std::vector<int64>& slice_limits() const;
|
| D | algebraic_simplifier.cc | 684 int64 slice_end = operands[i]->slice_limits(concatenate_dimension); in HandleConcatenate()
708 slice_end = operands[j]->slice_limits(concatenate_dimension); in HandleConcatenate()
716 auto new_limit_indices = operands[i]->slice_limits(); in HandleConcatenate()
2782 slice->slice_limits(i) != reshape->shape().dimensions(i)) { in TryToReorderSliceAndReshape()
2830 std::vector<int64> new_slice_limits = slice->slice_limits(); in HandleSlice()
2847 slice->slice_limits(dim) != in HandleSlice()
|
| D | dynamic_dimension_inference.cc | 400 hlo->slice_limits(dimension) != in HandleSlice()
|
| D | hlo_instructions.h | 623 int64 slice_limits(int64 dimension) const { return slice_limits_[dimension]; } in slice_limits() function
624 const std::vector<int64>& slice_limits() const { return slice_limits_; } in slice_limits() function
|
| D | hlo_verifier.cc | 572 slice->slice_limits(), slice->slice_strides())); in HandleSlice()
|
| D | algebraic_simplifier_test.cc | 2717 EXPECT_EQ(computation->root_instruction()->slice_limits(0), dim0 - 2); in TEST_F()
2718 EXPECT_EQ(computation->root_instruction()->slice_limits(1), dim1 - 4); in TEST_F()
4133 EXPECT_EQ(root->slice_limits(0), 2); in TEST_F()
|
| D | hlo_evaluator_typed_visitor.h | 2420 slice->slice_limits(), slice->slice_strides())); in HandleSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1989 ? slice->slice_limits(memcpy_dim) - slice->slice_starts(memcpy_dim) in HandleSlice()
|