/external/tensorflow/tensorflow/compiler/xla/tests/ |
slice_test.cc | 298 std::array<int64, 2> slice_limits; member in struct:xla::__anon39108::R2Spec 316 builder.Slice(a, spec.slice_starts, spec.slice_limits, spec.slice_strides); 319 input, spec.slice_starts, spec.slice_limits, spec.slice_strides); 379 std::array<int64, 4> slice_limits; member in struct:xla::__anon39108::R4Spec 389 "__limits_", Join(spec.slice_limits, "x"), // 402 values, spec.slice_starts, spec.slice_limits, spec.slice_strides); 409 builder.Slice(parameter, spec.slice_starts, spec.slice_limits,
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
hlo_constant_folding_test.cc | 149 const int64 slice_limits[] = {10, 8, 6, 5, 9}; local 158 shape, literal_instruction, slice_start, slice_limits, slice_strides));
|
hlo_instruction.h | 997 int64 slice_limits(int64 dimension) const { function in class:xla::HloInstruction 1001 const std::vector<int64>& slice_limits() const { function in class:xla::HloInstruction [all...] |
hlo_verifier.cc | 223 slice->slice_limits(), slice->slice_strides())); [all...] |
hlo_evaluator.cc | [all...] |
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
ir_emitter.cc | [all...] |