Searched defs:var_flat (Results 1 – 2 of 2) sorted by relevance
853 auto var_flat = var.flat_outer_dims<T>(); in DoCompute() local1039 auto var_flat = var.flat_outer_dims<T>(); in Compute() local1072 auto var_flat = var.flat<T>(); in Compute() local1388 auto var_flat = var.flat_outer_dims<T>(); in Compute() local1411 auto var_flat = var.flat<T>(); in Compute() local1546 auto var_flat = var.flat_outer_dims<T>(); in Compute() local1581 auto var_flat = var.flat<T>(); in Compute() local1842 auto var_flat = var.flat_outer_dims<T>(); in Compute() local1878 auto var_flat = var.flat<T>(); in Compute() local2205 auto var_flat = var.flat_outer_dims<T>(); in Compute() local[all …]
41 auto var_flat = var.flat<T1>(); in ReferenceBatchNorm() local104 auto var_flat = var.flat<T1>(); in FixedPointBatchNorm() local