Home
last modified time | relevance | path

Searched refs:CreateNestedSimpleLoops (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter_transforms.h39 std::vector<mlir::AffineForOp> CreateNestedSimpleLoops(
Dconv_emitter_transforms.cc66 std::vector<mlir::AffineForOp> CreateNestedSimpleLoops( in CreateNestedSimpleLoops() function
Dconv_emitter.cc259 CreateNestedSimpleLoops(output_shape_info.nchw_dimensions, builder); in CreateNaiveMlirConv()
274 reduction_loops = CreateNestedSimpleLoops( in CreateNaiveMlirConv()