Searched refs:CreateAndInitializeGradVariable (Results 1 – 1 of 1) sorted by relevance
378 LogicalResult CreateAndInitializeGradVariable(Type local_var_type, in CreateAndInitializeGradVariable() function411 if (failed(CreateAndInitializeGradVariable(local_var.getType(), grad, in HandleTensorArrayGradV3Op()615 if (failed(CreateAndInitializeGradVariable(operands[i].getType(), in HandleWhileOp()682 if (failed(CreateAndInitializeGradVariable(operands[i + 1].getType(), in HandleIfOp()742 if (failed(CreateAndInitializeGradVariable(it->getFirst().getType(), in HandlePartitionedCallOp()