Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc658 void GetOutputShapeForBroadcastGradientArgs(ArrayRef<int64_t> bcasted_shape, in GetOutputShapeForBroadcastGradientArgs() function
716 GetOutputShapeForBroadcastGradientArgs(bcasted_shape, s0_shape, s1_shape, r0, in verify()
750 GetOutputShapeForBroadcastGradientArgs(bcasted_shape, s0_shape, s1_shape, r0, in fold()