| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| strided_slice_op.cc | 57 Tensor begin_tensor, end_tensor, strides_tensor; variable
61 LiteralToHostTensor(end_literal, index_type_, &end_tensor));
69 &begin_tensor, &end_tensor, strides_tensor, input_shape,
140 Tensor begin_tensor, end_tensor, strides_tensor; variable
144 LiteralToHostTensor(end_literal, index_type_, &end_tensor));
151 &begin_tensor, &end_tensor, strides_tensor, input_shape,
248 Tensor begin_tensor, end_tensor, strides_tensor; variable
252 LiteralToHostTensor(end_literal, index_type_, &end_tensor));
266 &begin_tensor, &end_tensor, strides_tensor, lhs_shape,
|
| /external/tensorflow/tensorflow/core/util/ |
| strided_slice_op.cc | 34 const Tensor* end_tensor; member in struct:tensorflow::__anon40154::StridedSliceSparseSpec
88 dense->end_valid = sparse.end_tensor != nullptr;
119 if (sparse.end_tensor != nullptr) {
120 const auto& end_flat = sparse.end_tensor->flat<T>();
148 const Tensor* begin_tensor, const Tensor* end_tensor,
162 end_tensor != nullptr &&
163 !(TensorShapeUtils::IsVector(end_tensor->shape()) &&
164 end_tensor->NumElements() == strides_tensor.NumElements());
167 if (begin_tensor != nullptr && end_tensor != nullptr) {
171 end_tensor->shape().DebugString(), ", and "
[all...] |
