Home
last modified time | relevance | path

Searched refs:xla_params (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc376 llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromTfInner() argument
385 return mlir::BuildHloFromMlirHlo(block, builder, xla_params, returns); in BuildHloFromTfInner()
422 llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromTf() argument
436 TF_RETURN_IF_ERROR(BuildHloFromTfInner(module_op, builder, xla_params, in BuildHloFromTf()
617 llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromModule() argument
627 return BuildHloFromTf(module_op, builder, xla_params, returns, arg_shapes, in BuildHloFromModule()
679 llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromGraph() argument
692 return BuildHloFromModule(module.get(), builder, xla_params, returns, args, in BuildHloFromGraph()
Dcompile_mlir_util.h92 llvm::ArrayRef<xla::XlaOp> xla_params,
160 llvm::ArrayRef<xla::XlaOp> xla_params,
Dtf_xla_mlir_translate.cc255 std::vector<xla::XlaOp> xla_params; in CompileMlirToXlaHloViaBuilder() local
261 xla_params.push_back(argop); in CompileMlirToXlaHloViaBuilder()
265 TF_RETURN_IF_ERROR(BuildHloFromTf(module_op, builder, xla_params, returns, in CompileMlirToXlaHloViaBuilder()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate.cc142 std::vector<xla::XlaOp> xla_params; in ConvertMlirHloToHloViaBuilder() local
148 xla_params.push_back(argop); in ConvertMlirHloToHloViaBuilder()
153 mlir::BuildHloFromMlirHlo(block, builder, xla_params, returns, options)); in ConvertMlirHloToHloViaBuilder()
Dmlir_hlo_to_hlo.h59 llvm::ArrayRef<xla::XlaOp> xla_params,
Dmlir_hlo_to_hlo.cc1790 llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromMlirHlo() argument
1799 if (xla_params.size() != block.getArguments().size()) in BuildHloFromMlirHlo()
1804 lowering[arg] = xla_params[num]; in BuildHloFromMlirHlo()