Home
last modified time | relevance | path

Searched defs:TensorOrResourceShape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dcompile_mlir_util.h87 struct TensorOrResourceShape { struct
93 Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, argument
H A Dcompile_mlir_util.cc767 llvm::SmallVector<TensorOrResourceShape, 4>& arg_shapes) { in CompileGraphSetup()