Searched defs:accum_shape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | training_ops.cc | 126 TensorShape var_shape, accum_shape; in Compile() local
184 TensorShape var_shape, accum_shape; in Compile() local
242 TensorShape var_shape, accum_shape; in Compile() local
291 TensorShape var_shape, accum_shape; in Compile() local
344 TensorShape var_shape, accum_shape; in Compile() local
400 TensorShape var_shape, accum_shape, squared_accum_shape; in Compile() local
754 TensorShape var_shape, accum_shape, linear_shape; in CompileFtrl() local
915 TensorShape var_shape, accum_shape, accum_update_shape; in Compile() local
|