Searched refs:axis_input (Results 1 – 1 of 1) sorted by relevance
151 const auto& axis_input = properties.GetInputProperties(node->name()).back(); in GetConcatAxis() local152 if (!TensorShape::IsValid(axis_input.shape()) || !axis_input.has_value()) { in GetConcatAxis()156 Tensor axis_tensor(axis_input.dtype(), axis_input.shape()); in GetConcatAxis()157 if (!axis_tensor.FromProto(axis_input.value())) { in GetConcatAxis()160 *axis = axis_input.dtype() == DT_INT64 in GetConcatAxis()