Searched refs:algorithm_attr (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | mkldnn_fusion.py | 250 unary_attr.algorithm_attr,
491 unary_attr.algorithm_attr,
535 unary_attr.algorithm_attr,
555 self, op_name: str, scalars_attr=None, algorithm_attr=None argument
559 self.algorithm_attr = algorithm_attr if algorithm_attr else ""
566 UnaryAttr("gelu", algorithm_attr="tanh"): [
570 UnaryAttr("gelu", algorithm_attr="none"): [
|
| H A D | quantization.py | 367 unary_attr.algorithm_attr,
441 unary_attr.algorithm_attr,
522 binary_unary_attr.algorithm_attr,
689 binary_unary_attr.algorithm_attr,
709 self, op_name: str, scalars_attr=None, algorithm_attr=None argument
713 self.algorithm_attr = algorithm_attr if algorithm_attr else ""
905 algorithm_attr=None, argument
911 self.algorithm_attr = algorithm_attr if algorithm_attr else ""
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | model_dataset_op.cc | 108 AttrValue algorithm_attr; in AsGraphDefInternal() local
109 b->BuildAttrValue(static_cast<int64_t>(algorithm_), &algorithm_attr); in AsGraphDefInternal()
117 {std::make_pair(kAlgorithm, algorithm_attr), in AsGraphDefInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 467 auto algorithm_attr = mlir::mhlo::RngAlgorithmAttr::get( in RngBitGeneratorInternal() local
470 loc_, flattened_ret_types, algorithm_attr, GetValue(initial_state)); in RngBitGeneratorInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| H A D | lmhlo_to_jitrt.cc | 441 auto algorithm_attr = op.getAlgorithm() in matchAndRewrite() local
444 call->setAttr(b.getStringAttr("algorithm"), algorithm_attr); in matchAndRewrite()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | hlo_function_importer.cc | 1100 auto algorithm_attr = mlir::mhlo::RngAlgorithmAttr::get( in ImportInstructionImpl() local
1104 loc, flattened_ret_types, algorithm_attr, operands[0]); in ImportInstructionImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 7161 auto algorithm_attr = mlir::mhlo::RngAlgorithmAttr::get( in matchAndRewrite() local
7165 loc, op.getResultTypes(), algorithm_attr, op.initial_state()); in matchAndRewrite()
|