Home
last modified time | relevance | path

Searched refs:TensorOrResourceShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.h79 struct TensorOrResourceShape { struct
85 Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, argument
94 llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
103 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
112 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
Dcompile_mlir_util.cc83 mlir::ModuleOp module, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in GetXlaInputShapes()
228 Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in RefineShapes()
239 for (const TensorOrResourceShape& tensor_resource_shape : arg_shapes) { in RefineShapes()
246 for (const TensorOrResourceShape& tensor_resource_shape : arg_shapes) { in RefineShapes()
407 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirSetup()
424 llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in BuildHloFromTf()
447 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in PopulateResultIOInfo()
468 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHlo()
504 llvm::SmallVector<TensorOrResourceShape, 4> tensor_or_resource_shapes; in CompileSerializedMlirToXlaHlo()
590 llvm::SmallVector<TensorOrResourceShape, 4>& arg_shapes) { in CompileGraphSetup() argument
[all …]
Dtf_xla_mlir_translate.cc119 llvm::SmallVectorImpl<TensorOrResourceShape>& arg_shapes) { in ParseArgumentShapes()
241 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHloViaBuilder()
297 llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes; in MlirTfToHloTextTranslateFunctionImpl()