Searched refs:output_shape_info (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 250 const ShapeInfo& output_shape_info, const Window& window, in CreateNaiveMlirConv() argument
254 CHECK(output_shape_info.element_type == builder.getF16Type()); in CreateNaiveMlirConv()
259 CreateNestedSimpleLoops(output_shape_info.nchw_dimensions, builder); in CreateNaiveMlirConv()
366 output, output_shape_info.affine_map, output_vars); in CreateNaiveMlirConv()
545 ShapeInfo output_shape_info = GetShapeInfo( in EmitConvolutionForwardAsMlir() local
554 {mlir::MemRefType::get(output_shape_info.physical_dimensions, in EmitConvolutionForwardAsMlir()
555 output_shape_info.element_type, {}), in EmitConvolutionForwardAsMlir()
576 filter_shape_info, output_shape_info, conv->window(), in EmitConvolutionForwardAsMlir()
|