Searched defs:max_op (Results 1 – 9 of 9) sorted by relevance
55 MaxOp<float> max_op; in reduce_max() local99 MaxOp<float> max_op; in max_scale_exp2_sum() local
1147 MaxOp<float> max_op; in combine_attn_seqk_parallel() local
51 const auto* max_op = op_0->type == OperatorType::kMaximum ? op_0 : op_1; in Run() local
369 } max_op; in compute_mode() local
494 CustomMax max_op{}; in _segment_reduce_lengths_offsets_cuda_kernel() local
776 uint64_t max_op = MaxOp(); in Build() local812 uint64_t max_op = 0; in MaxOp() local
824 uint64_t max_op = MaxOp(); in Build() local868 mutable uint64_t max_op = 0; member
768 tensorflow::NodeDef* max_op = tensorflow_graph->add_node(); in ConvertRelu1Operator() local1903 tensorflow::NodeDef* max_op = tensorflow_graph->add_node(); in ConvertTensorFlowMaximumOperator() local
4618 auto max_op = builder.create<MaxOp>(loc, block->getArgument(0), in BuildScatterBody() local