Searched refs:XlaArgument (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_argument.cc | 22 bool XlaArgument::operator==(const XlaArgument& other) const { in operator ==()
55 bool AnyUninitializedResourceArg(absl::Span<const XlaArgument> args) { in AnyUninitializedResourceArg()
56 return llvm::any_of(args, [](const XlaArgument& arg) { in AnyUninitializedResourceArg()
57 return arg.kind == XlaArgument::kResource && arg.type == DT_INVALID; in AnyUninitializedResourceArg()
|
| D | xla_argument.h | 31 struct XlaArgument { struct
113 bool operator==(const XlaArgument& other) const; argument
127 bool AnyUninitializedResourceArg(absl::Span<const XlaArgument> args);
|
| D | xla_compiler.h | 102 using Argument = ::tensorflow::XlaArgument;
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tf_xla_mlir_translate.cc | 162 llvm::SmallVectorImpl<XlaArgument::Kind>& argument_kinds) { in ParseArgumentKinds()
172 argument_kinds.push_back(XlaArgument::Kind::kParameter); in ParseArgumentKinds()
174 argument_kinds.push_back(XlaArgument::Kind::kResource); in ParseArgumentKinds()
188 llvm::SmallVectorImpl<XlaArgument>& xla_arguments) { in ParseXlaArguments()
195 llvm::SmallVector<XlaArgument::Kind, 4> arg_kinds_vector; in ParseXlaArguments()
203 XlaArgument::Kind::kParameter); in ParseXlaArguments()
217 XlaArgument& arg = std::get<0>(arg_components); in ParseXlaArguments()
333 llvm::SmallVector<XlaArgument, 4> xla_arguments; in MlirTfGraphToHloTextTranslateFunction()
|
| D | compile_mlir_util.cc | 70 StatusOr<TensorShape> GetTensorShapeFromXlaArgument(const XlaArgument& arg) { in GetTensorShapeFromXlaArgument()
519 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args) { in RewriteWithArgs()
527 const XlaArgument& xla_arg = args[idx]; in RewriteWithArgs()
529 if (xla_arg.kind == XlaArgument::kResource) { in RewriteWithArgs()
557 if (xla_arg.kind != XlaArgument::kConstant) { in RewriteWithArgs()
588 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args, in CompileGraphSetup()
598 /*is_resource=*/arg.kind == XlaArgument::kResource}); in CompileGraphSetup()
619 llvm::ArrayRef<XlaArgument> args, in BuildHloFromModule()
632 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args, in CompileGraphToXlaHlo()
681 llvm::ArrayRef<XlaArgument> args, in BuildHloFromGraph()
[all …]
|
| D | compile_mlir_util.h | 136 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args,
146 const Graph& graph, llvm::ArrayRef<XlaArgument> args,
162 llvm::ArrayRef<XlaArgument> args,
|