Searched refs:tensor_or_resource_shapes (Results 1 – 1 of 1) sorted by relevance
504 llvm::SmallVector<TensorOrResourceShape, 4> tensor_or_resource_shapes; in CompileSerializedMlirToXlaHlo() local505 tensor_or_resource_shapes.reserve(arg_shapes.size()); in CompileSerializedMlirToXlaHlo()507 tensor_or_resource_shapes.push_back({arg_shape}); in CompileSerializedMlirToXlaHlo()509 mlir_module.get(), tensor_or_resource_shapes, device_type, use_tuple_args, in CompileSerializedMlirToXlaHlo()