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
|