Searched refs:assn (Results 1 – 2 of 2) sorted by relevance
416 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall() local417 auto output_data = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie(); in HandleCustomCall()418 auto output_mean = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie(); in HandleCustomCall()419 auto output_inv_stddev = assn.GetUniqueSlice(custom_call, {2}).ValueOrDie(); in HandleCustomCall()446 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall() local447 auto output_grad_data = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie(); in HandleCustomCall()448 auto output_grad_scale = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie(); in HandleCustomCall()450 assn.GetUniqueSlice(custom_call, {2}).ValueOrDie(); in HandleCustomCall()468 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall() local475 auto conv_result_slice = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie(); in HandleCustomCall()[all …]
11638 assn %17192