Home
last modified time | relevance | path

Searched defs:min_op (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dvectorization_pass.cc109 auto is_op_of_interest = [&](AffineMinOp min_op, Value iv) { in SplitSCFForOp()
138 scf_for->walk([&](AffineMinOp min_op) { in SplitSCFForOp()
159 new_loop->walk([&](AffineMinOp min_op) { in SplitSCFForOp()
180 tail_if->walk([&](AffineMinOp min_op) { in SplitSCFForOp()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_relu1.cc50 const auto* min_op = op_0->type == OperatorType::kMinimum ? op_0 : op_1; in Run() local
/external/pytorch/aten/src/ATen/native/cuda/
DSegmentReduce.cu530 CustomMin min_op{}; in _segment_reduce_lengths_offsets_cuda_kernel() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc775 tensorflow::NodeDef* min_op = tensorflow_graph->add_node(); in ConvertRelu1Operator() local
1889 tensorflow::NodeDef* min_op = tensorflow_graph->add_node(); in ConvertTensorFlowMinimumOperator() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4600 auto min_op = builder.create<MinOp>(loc, block->getArgument(0), in BuildScatterBody() local