Searched defs:grad_flat (Results 1 – 3 of 3) sorted by relevance
1246 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1289 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1332 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1377 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1422 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1466 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1511 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1556 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1601 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1652 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local[all …]
1263 auto grad_flat = grad.flat_outer_dims<T>(); in DoCompute() local1445 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local1478 auto grad_flat = grad.flat<T>(); in Compute() local2462 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local2497 auto grad_flat = grad.flat<T>(); in Compute() local3206 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local4274 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local4413 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local
266 auto grad_flat = grad_val->flat_outer_dims<T>(); in AddToAccumGradFunction() local