Searched defs:tf_shape (Results 1 – 5 of 5) sorted by relevance
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