Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dmlir_xla_op_kernel.cc72 llvm::SmallVector<xla::XlaOp, 4> xla_params(ctx->num_inputs()); in ConstructXlaOp() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc502 llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromTfInner()
565 llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromTf()
802 llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromModule()
862 mlir::MLIRContext& mlir_context, llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromGraph()
Dtf_xla_mlir_translate.cc269 std::vector<xla::XlaOp> xla_params; in CompileMlirToXlaHloViaBuilder() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate_registration.cc112 std::vector<xla::XlaOp> xla_params; in ConvertMlirHloToHloViaBuilder() local
Dmlir_hlo_to_hlo.cc2580 llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromMlirHlo()