Home
last modified time | relevance | path

Searched refs:MakeShapeFromTensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner.cc505 target_context->MakeShapeFromTensor(&t, src_shape, result).ok()) { in ConstantPartialShape()
516 target_context->MakeShapeFromTensor(nullptr, src_shape, result)); in ConstantPartialShape()
589 TF_RETURN_IF_ERROR(target_context->MakeShapeFromTensor( in ConstantPartialShape()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.h711 Status MakeShapeFromTensor(const Tensor* t, ShapeHandle tensor_shape,
H A Dshape_inference.cc762 Status InferenceContext::MakeShapeFromTensor(const Tensor* t, in MakeShapeFromTensor() function in tensorflow::shape_inference::InferenceContext