Searched refs:begin_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 33 const Tensor* begin_tensor; member
87 dense->begin_valid = sparse.begin_tensor != nullptr; in BuildDenseSpec()
90 const T* const begin_flat = sparse.begin_tensor != nullptr in BuildDenseSpec()
91 ? sparse.begin_tensor->vec<T>().data() in BuildDenseSpec()
153 const Tensor* begin_tensor, const Tensor* end_tensor, in ValidateStridedSliceOp() argument
162 begin_tensor != nullptr && in ValidateStridedSliceOp()
163 !(TensorShapeUtils::IsVector(begin_tensor->shape()) && in ValidateStridedSliceOp()
164 begin_tensor->NumElements() == strides_tensor.NumElements() && in ValidateStridedSliceOp()
165 begin_tensor->NumElements() < 32 /* using 32 bit masks */); in ValidateStridedSliceOp()
172 if (begin_tensor != nullptr && end_tensor != nullptr) { in ValidateStridedSliceOp()
[all …]
|
| D | strided_slice_op.h | 44 const Tensor* begin_tensor, const Tensor* end_tensor,
55 const Tensor* begin_tensor, const Tensor* end_tensor,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 58 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
60 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
70 &begin_tensor, &end_tensor, strides_tensor, input_shape, in Compile()
142 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
144 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
153 &begin_tensor, &end_tensor, strides_tensor, input_shape, in Compile()
250 Tensor begin_tensor, end_tensor, strides_tensor; in Compile() local
252 ctx, LiteralToHostTensor(begin_literal, index_type_, &begin_tensor)); in Compile()
268 &begin_tensor, &end_tensor, strides_tensor, lhs_shape, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 71 const Tensor& begin_tensor = context->input(1); in SharedValidation() local
76 context->op_kernel().IsLegacyVector(begin_tensor.shape()) && in SharedValidation()
78 begin_tensor.NumElements() == input.dims() && in SharedValidation()
82 input.dims(), ", but got shapes ", begin_tensor.shape().DebugString(), in SharedValidation()
86 *begin = IntTensorToInt64Vec(begin_tensor); in SharedValidation()
|
| D | mkl_slice_op.cc | 73 const Tensor& begin_tensor = MklGetInput(context, kInputBeginIndex); in ValidateMklInputs() local
91 context, context->op_kernel().IsLegacyVector(begin_tensor.shape()) && in ValidateMklInputs()
93 begin_tensor.NumElements() == input_dims && in ValidateMklInputs()
97 input_dims, ", but got shapes ", begin_tensor.shape().DebugString(), in ValidateMklInputs()
100 *begin = IntTensorToInt64Vec(begin_tensor); in ValidateMklInputs()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 952 Tensor begin_tensor; in TEST_F() local
953 EXPECT_TRUE(begin_tensor.FromProto( in TEST_F()
957 test::ExpectTensorEqual<int>(begin_tensor_expected, begin_tensor); in TEST_F()
|