Searched refs:inference_shape_ (Results 1 – 2 of 2) sorted by relevance
305 if (!IsReady() && inference_shape_.IsFullyDefined()) { in Shape()306 bool fill = inference_shape_.AsTensorShape(shape); in Shape()331 if (inference_shape_.unknown_rank()) { in InferenceShape()336 inference_shape_.dims()); in InferenceShape()338 dims_handle[i] = inference_context->MakeDim(inference_shape_.dim_size(i)); in InferenceShape()351 inference_shape_ = PartialTensorShape(); in SetInferenceShape()360 &inference_shape_); in SetInferenceShape()372 inference_shape_ = other_shape; in CopyInferenceShape()374 inference_shape_ = other->inference_shape_; in CopyInferenceShape()381 if (!IsReady() && !inference_shape_.unknown_rank()) { in NumDims()[all …]
286 PartialTensorShape inference_shape_; variable