| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| dynamic_ops_test.cc | 123 const std::vector<IndexT> slice_starts,
144 slice_starts, 0, "slice_starts", &builder, &starts);
154 const std::vector<IndexT> slice_starts,
170 slice_starts, 0, "slice_starts", &builder, &starts);
180 const std::vector<IndexT> slice_starts,
196 slice_starts, 0, "slice_starts", &builder, &starts);
347 const std::vector<IndexT> slice_starts,
[all...] |
| slice_test.cc | 297 std::array<int64, 2> slice_starts; 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);
378 std::array<int64, 4> slice_starts; member in struct:xla::__anon39108::R4Spec
388 "__starts_", Join(spec.slice_starts, "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_instruction.h | 987 int64 slice_starts(int64 dimension) const { function in class:xla::HloInstruction
991 const std::vector<int64>& slice_starts() const { return slice_starts_; } function in class:xla::HloInstruction
[all...] |
| hlo_verifier.cc | 222 slice->operand(0)->shape(), slice->slice_starts(),
[all...] |
| hlo_evaluator.cc | [all...] |
| elemental_ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | [all...] |
