Home
last modified time | relevance | path

Searched defs:TensorOrResourceShape (Results 1 – 2 of 2) 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
Dcompile_mlir_util.cc590 llvm::SmallVector<TensorOrResourceShape, 4>& arg_shapes) { in CompileGraphSetup()