Searched refs:inference_shape_ (Results 1 – 2 of 2) sorted by relevance
534 if (!IsReady() && inference_shape_.IsFullyDefined()) { in Shape()535 bool fill = inference_shape_.AsTensorShape(shape); in Shape()560 if (inference_shape_.unknown_rank()) { in InferenceShape()565 inference_shape_.dims()); in InferenceShape()567 dims_handle[i] = inference_context->MakeDim(inference_shape_.dim_size(i)); in InferenceShape()580 inference_shape_ = PartialTensorShape(); in SetInferenceShape()589 &inference_shape_); in SetInferenceShape()601 inference_shape_ = other_shape; in CopyInferenceShape()603 inference_shape_ = other->inference_shape_; in CopyInferenceShape()610 if (!IsReady() && !inference_shape_.unknown_rank()) { in Shape()[all …]
388 PartialTensorShape inference_shape_; variable