Searched defs:GetTensorDim (Results 1 – 1 of 1) sorted by relevance
425 T GetTensorDim(gtl::ArraySlice<T> dimension_attributes, in GetTensorDim() function448 T GetTensorDim(const std::vector<T>& attributes, TensorFormat format, in GetTensorDim() function455 inline int64 GetTensorDim(const TensorShape& tensor_shape, in GetTensorDim() function472 inline int64 GetTensorDim(const Tensor& tensor, TensorFormat tensor_format, in GetTensorDim() function