Searched defs:grad_flat (Results 1 – 3 of 3) sorted by relevance
1406 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1449 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1492 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1537 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1582 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1626 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1671 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1716 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1761 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local1812 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() local[all …]
1485 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local1518 auto grad_flat = grad.flat<T>(); in Compute() local2494 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local2529 auto grad_flat = grad.flat<T>(); in Compute() local3230 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local4270 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local4409 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() local
245 auto grad_flat = grad_val->flat_outer_dims<T>(); in AddToAccumGradFunction() local