Searched refs:saved_variance_tensor (Results 1 – 1 of 1) sorted by relevance
744 Tensor* saved_variance_tensor = nullptr; in Compute() local793 &saved_mean_tensor, &saved_variance_tensor, in Compute()805 &saved_mean_tensor, &saved_variance_tensor, in Compute()829 reinterpret_cast<char*>(saved_variance_tensor->flat<U>().data()); in Compute()864 U* variance_op_data = saved_variance_tensor->flat<U>().data(); in Compute()956 Tensor* saved_variance_tensor = nullptr; in HandleEmptyInput() local960 &saved_mean_tensor, &saved_variance_tensor, in HandleEmptyInput()969 Tensor** saved_variance_tensor, in AllocateTFOutputs() argument974 DCHECK(saved_variance_tensor); in AllocateTFOutputs()1022 saved_variance_tensor, tf_shape_scale, in AllocateTFOutputs()[all …]