Searched refs:end_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 36 const Tensor* end_tensor; member
97 dense->end_valid = sparse.end_tensor != nullptr; in BuildDenseSpec()
102 const T* const end_flat = sparse.end_tensor != nullptr in BuildDenseSpec()
103 ? sparse.end_tensor->vec<T>().data() in BuildDenseSpec()
170 const Tensor* begin_tensor, const Tensor* end_tensor, in ValidateStridedSliceOp() argument
184 end_tensor != nullptr && in ValidateStridedSliceOp()
185 !(TensorShapeUtils::IsVector(end_tensor->shape()) && in ValidateStridedSliceOp()
186 end_tensor->NumElements() == strides_tensor.NumElements()); in ValidateStridedSliceOp()
189 if (begin_tensor != nullptr && end_tensor != nullptr) { in ValidateStridedSliceOp()
193 end_tensor->shape().DebugString(), ", and ", in ValidateStridedSliceOp()
[all …]
|
| D | strided_slice_op.h | 65 const Tensor* begin_tensor, const Tensor* end_tensor,
77 const Tensor* begin_tensor, const Tensor* end_tensor,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 214 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
221 ctx, LiteralToHostTensor(end_literal, index_type_, &end_tensor)); in Compile()
233 end_is_constant ? &end_tensor : nullptr, strides_tensor, in Compile()
495 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
499 LiteralToHostTensor(end_literal, index_type_, &end_tensor)); in Compile()
506 &begin_tensor, &end_tensor, strides_tensor, input_shape, in Compile()
627 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
631 LiteralToHostTensor(end_literal, index_type_, &end_tensor)); in Compile()
650 &begin_tensor, &end_tensor, strides_tensor, lhs_shape, in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | strided_slice_builder.cc | 35 const auto& end_tensor = context->tensors[inputs->data[2]]; in PopulateSubGraph() local
40 graph_builder_->AddConstNodeWithData(inputs->data[2], end_tensor); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 170 auto end_tensor = CreateI32DenseConst(builder, end, loc); in CreateStridedSliceOp() local
174 loc, output_type, input, begin_tensor, end_tensor, strides_tensor, in CreateStridedSliceOp()
|