Searched refs:input_grad_tensor (Results 1 – 1 of 1) sorted by relevance
372 const Tensor& input_grad_tensor = MklGetInput(context, kIdxGradient); in Compute() local380 input_grad_tensor, input_grad_dnn_shape, &input_grad_dnn_data); in Compute()477 memory::desc ConfigureInputGradient(const Tensor& input_grad_tensor, in ConfigureInputGradient() argument486 input_grad_dnn_data->SetUsrMem(input_grad_md, &input_grad_tensor); in ConfigureInputGradient()