Searched refs:input_gradient_tensor (Results 1 – 1 of 1) sorted by relevance
504 Tensor input_gradient_tensor; in MklDefaultToEigen() local518 input_grad_dnn_shape, &input_gradient_tensor)); in MklDefaultToEigen()520 input_gradient_tensor = MklGetInput(context, kIdxGradient); in MklDefaultToEigen()539 const int64 batch = static_cast<int64_t>(input_gradient_tensor.dim_size(0)); in MklDefaultToEigen()540 const int64 rows = static_cast<int64_t>(input_gradient_tensor.dim_size(1)); in MklDefaultToEigen()541 const int64 cols = static_cast<int64_t>(input_gradient_tensor.dim_size(2)); in MklDefaultToEigen()542 const int64 depth = static_cast<int64_t>(input_gradient_tensor.dim_size(3)); in MklDefaultToEigen()546 input_gradient_tensor.shaped<T, 2>({nodes * batch, depth}); in MklDefaultToEigen()556 input_gradient_tensor.shape(), in MklDefaultToEigen()592 const Tensor& input_gradient_tensor = MklGetInput(context, kIdxGradient); in SanityCheckInputs() local[all …]