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