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