Home
last modified time | relevance | path

Searched refs:concat_dim_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_op.cc62 const Tensor& concat_dim_tensor = c->input(axis_input_index_); in Compute() local
66 (TensorShapeUtils::IsScalar(concat_dim_tensor.shape()) || in Compute()
67 (TensorShapeUtils::IsVector(concat_dim_tensor.shape()) && in Compute()
68 concat_dim_tensor.shape().dim_size(0) == 1)), in Compute()
72 concat_dim_tensor.shape().DebugString())); in Compute()
78 (concat_dim_tensor.dtype() == DT_INT32 || in Compute()
79 concat_dim_tensor.dtype() == DT_INT64), in Compute()
82 DataTypeString(concat_dim_tensor.dtype()))); in Compute()
84 OP_REQUIRES(c, (concat_dim_tensor.dtype() == DT_INT32), in Compute()
87 DataTypeString(concat_dim_tensor.dtype()))); in Compute()
[all …]
H A Dquantized_concat_op.cc161 const Tensor* concat_dim_tensor = nullptr; in Compute() local
162 OP_REQUIRES_OK(context, context->input("concat_dim", &concat_dim_tensor)); in Compute()
164 context, TensorShapeUtils::IsScalar(concat_dim_tensor->shape()), in Compute()
167 concat_dim_tensor->shape().DebugString())); in Compute()
168 const int32_t concat_dim = concat_dim_tensor->scalar<int32>()(); in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_concat_op.cc140 const Tensor* concat_dim_tensor; in Compute() local
145 OP_REQUIRES_OK(c, c->input(axis_attribute_name, &concat_dim_tensor)); in Compute()
146 OP_REQUIRES(c, TensorShapeUtils::IsScalar(concat_dim_tensor->shape()), in Compute()
150 concat_dim_tensor->shape().DebugString())); in Compute()
152 internal::SubtleMustCopy(concat_dim_tensor->scalar<int32>()()); in Compute()
476 const Tensor& concat_dim_tensor = (AxisArgName == NAME_IS_CONCAT_DIM) in Compute() local
481 context, TensorShapeUtils::IsScalar(concat_dim_tensor.shape()), in Compute()
484 concat_dim_tensor.shape().DebugString())); in Compute()
486 internal::SubtleMustCopy(concat_dim_tensor.scalar<int32>()()); in Compute()