Home
last modified time | relevance | path

Searched refs:BroadcastBinaryOpOutputShapeFnHelper (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dstateless_random_ops.cc148 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in __anonbacd0d810302()
150 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in __anonbacd0d810302()
152 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in __anonbacd0d810302()
H A Dlinalg_ops.cc78 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in BandedTriangularSolveShapeFn()
144 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in MatrixTriangularSolveShapeFn()
H A Dmath_ops.cc755 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( \
936 BroadcastBinaryOpOutputShapeFnHelper(c, then, else_, true, &other)); in __anon247ac1530502()
939 BroadcastBinaryOpOutputShapeFnHelper(c, cond, other, true, &output)); in __anon247ac1530502()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.h226 Status BroadcastBinaryOpOutputShapeFnHelper(InferenceContext* c,
237 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in BroadcastBinaryOpOutputShapeFn()
H A Dcommon_shape_fns.cc258 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in EinsumShape()
335 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in BatchMatMulV2Shape()
2082 Status BroadcastBinaryOpOutputShapeFnHelper(InferenceContext* c, in BroadcastBinaryOpOutputShapeFnHelper() function