Searched refs:is_grad (Results 1 – 2 of 2) sorted by relevance
141 bool multiple_writes_aggregate, bool is_grad, int32 marked_size, in TensorArray() argument151 is_grad_(is_grad), in TensorArray()
466 OpContext DescribeFusedBatchNorm(const bool is_training, const bool is_grad, in DescribeFusedBatchNorm() argument474 if (is_grad) { in DescribeFusedBatchNorm()481 if (is_grad) { in DescribeFusedBatchNorm()484 int num_1d_inputs = is_grad ? 3 : 4; in DescribeFusedBatchNorm()