Searched refs:get_layout_attribute (Results 1 – 1 of 1) sorted by relevance
733 auto get_layout_attribute = [&](const xla::Layout& layout) { in EmitDnnConvolution() local783 get_layout_attribute(custom_call->operand(0)->shape().layout()), in EmitDnnConvolution()784 get_layout_attribute(custom_call->operand(1)->shape().layout()), in EmitDnnConvolution()785 get_layout_attribute(custom_call->shape().tuple_shapes(0).layout()), in EmitDnnConvolution()