Home
last modified time | relevance | path

Searched defs:var_flat (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_batch_norm_op.cc41 auto var_flat = var.flat<T1>(); in ReferenceBatchNorm() local
104 auto var_flat = var.flat<T1>(); in FixedPointBatchNorm() local
H A Dtraining_ops.cc1484 auto var_flat = var.flat_outer_dims<T>(); in Compute() local
1517 auto var_flat = var.flat<T>(); in Compute() local
2490 auto var_flat = var.flat_outer_dims<T>(); in Compute() local
2526 auto var_flat = var.flat<T>(); in Compute() local
3228 auto var_flat = var.flat_outer_dims<T>(); in Compute() local
4267 auto var_flat = var.flat_outer_dims<T>(); in Compute() local
4405 auto var_flat = var.flat_outer_dims<T>(); in Compute() local