Searched refs:BroadcastBinaryOpOutputShapeFnHelper (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
H A D | stateless_random_ops.cc | 148 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 D | linalg_ops.cc | 78 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in BandedTriangularSolveShapeFn() 144 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in MatrixTriangularSolveShapeFn()
|
H A D | math_ops.cc | 755 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 D | common_shape_fns.h | 226 Status BroadcastBinaryOpOutputShapeFnHelper(InferenceContext* c, 237 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in BroadcastBinaryOpOutputShapeFn()
|
H A D | common_shape_fns.cc | 258 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in EinsumShape() 335 TF_RETURN_IF_ERROR(BroadcastBinaryOpOutputShapeFnHelper( in BatchMatMulV2Shape() 2082 Status BroadcastBinaryOpOutputShapeFnHelper(InferenceContext* c, in BroadcastBinaryOpOutputShapeFnHelper() function
|