| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| strided_slice_op.cc | 75 gtl::InlinedVector<int64, 4> slice_begin, slice_end, slice_strides; variable
81 slice_strides.push_back(strides[i]);
87 slice_strides.push_back(-strides[i]);
97 slice = ctx->builder()->Slice(slice, slice_begin, slice_end, slice_strides);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| slice_test.cc | 299 std::array<int64, 2> slice_strides; 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);
380 std::array<int64, 4> slice_strides; member in struct:xla::__anon39108::R4Spec
390 "__strides_", Join(spec.slice_strides, "x") //
402 values, spec.slice_starts, spec.slice_limits, spec.slice_strides);
410 spec.slice_strides);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_constant_folding_test.cc | 150 const int64 slice_strides[] = {1, 1, 1, 1, 1}; local
158 shape, literal_instruction, slice_start, slice_limits, slice_strides));
|
| hlo_instruction.h | 1009 int64 slice_strides(int64 dimension) const { function in class:xla::HloInstruction
1013 const std::vector<int64>& slice_strides() const { return slice_strides_; } function in class:xla::HloInstruction
[all...] |
| hlo_verifier.cc | 223 slice->slice_limits(), slice->slice_strides()));
[all...] |
| hlo_evaluator.cc | [all...] |
| elemental_ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | [all...] |
