Searched refs:algorithm_attr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| D | lmhlo_to_jitrt.cc | 441 auto algorithm_attr = op.getAlgorithm() in matchAndRewrite() local
444 call->setAttr(b.getStringAttr("algorithm"), algorithm_attr); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| 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()
|