Searched defs:tf_shape (Results 1 – 7 of 7) sorted by relevance
37 tf::TensorShape tf_shape; in ConvertShape() local78 tf::TensorShape tf_shape = ConvertShape(tensor.shape()); in Add() local
66 tf::TensorShape tf_shape; in ConvertTensorSpec() local
73 tensorflow::TensorShape tf_shape = GetTfShape(shape); in TEST() local
550 tensorflow::PartialTensorShape tf_shape(shape_attr.GetShape()); in SetUpScalarAttr() local587 tensorflow::PartialTensorShape tf_shape(shape.GetShape()); in AddShapeToAttrList() local
129 std::vector<::tensorflow::shape_inference::DimensionHandle> tf_shape; in SetOutputShape() local
108 auto tf_shape = shape.dim_sizes(); in ConvertRuntimeFallbackTensorToDenseGpuTensor() local
791 TensorShape tf_shape; in AllocTmpBuffer() local801 TensorShape tf_shape) { in AllocTmpBuffer()819 TensorShape tf_shape; in AllocateSPTensor() local